Python Pandas從.csv提取列

Question

我有一個.csv文件，我可以在Pandas中讀取它。 .csv文件如下所示。

a b c d e 
1 4 3 2 5 
6 7 8 3 6 
... 

我需要實現的是，我可以提取和b作爲列向量和 並[c d E]作爲基質。我用熊貓用下面的代碼讀取.csv文件：

pd.read_csv（「data.csv」，九月=「」，標題=無）

但是這會給我像一個矢量這：[[a，b，c，d，e]，[1,4,3,2,5]，...]

如何提取列？我聽說過df.iloc，但這裏不能使用，因爲在pd.read_csv之後只有一列。

Answer 1

你應該能夠做到這一點與：

ds = pd.read_csv('data.csv', sep=',',header=0) 
column_a = ds["a"] 
matrix = ds[["c","d","e"]]