我有一个.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之后只有一列。
答案 0 :(得分:0)
您应该能够:
ds = pd.read_csv('data.csv', sep=',',header=0)
column_a = ds["a"]
matrix = ds[["c","d","e"]]