Python Pandas从.csv中提取列

时间:2017-03-17 08:31:14

标签: pandas

我有一个.csv文件,我可以在Pandas中阅读。 .csv文件如下所示。

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

我需要实现的是,我可以提取a和b作为列向量和 [c d e]作为矩阵。我使用以下代码的Pandas来读取.csv文件:

pd.read_csv(' data.csv',sep =',',header =无)

但这会给我一个这样的矢量:[[a,b,c,d,e],[1,4,3,2,5],...]

如何提取列?我听说过df.iloc,但是这里不能使用,因为在pd.read_csv之后只有一列。

1 个答案:

答案 0 :(得分:0)

您应该能够:

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