python - 使用熊猫将csv文件中的列范围分配给变量

Question

我有一个包含 30 列的 CSV 文件，我有两个变量 (X,y) 我希望变量 y 从列号 0 中获取所有行，所以我使用

df = pd.read_csv("dataset.csv" )
y = df[:0]

这只会得到我不需要的标题

我希望变量 X 从第 4 列到第 23 列中获取所有行

X = df[4:23] 

但这不起作用，因为它从第 4 行获取所有列，所以它与我需要的相反。

标签： pythonpandasdataframecsv

Answer 1

我只是偶然发现了我应该使用 iloc 的答案，所以应该是

y = df.iloc[: , 0].values
X = df.iloc[ :  , 4: 23 ].values