Python 读取 csv 文件中的 3 列

Question

我想从 .csv 文件中读取列。

这是我试过的代码:

inputFileName = "some.csv"
dataset = pd.read_csv(inputFileName)
dataset = np.array(dataset)
ihr = np.array(dataset[:])
x = dataset[:, 0]
y = dataset[:,1]

该代码的目的是从some.csv 中读取第一列值和第二列值，并将其分别存储为x 和y。但是，它没有这样做。

当我执行这个时，这就是我得到的:

['2.070\t72.2892\t0' '2.900\t72.2892\t0' '3.730\t68.1818\t0' ...,
 '29562.020\t74.0741\t0' '29562.830\t75\t0' '29563.630\t73.1707\t0']

Traceback (most recent call last):
  File "/home/ubuntu/Desktop/major_thesis/ECG/transpose_ihr.py", line 32, in <module>
    y = dataset[:, 1]
IndexError: index 1 is out of bounds for axis 1 with size 1

此代码的问题在于它读取整个数据文件并将其存储为数据集。那么，如果我想单独阅读这两列，我该怎么办

提前致谢

Answer 1

我认为你有 separator tab，所以需要更改默认值 sep=','(id 没有 sep 参数)到 read_csv 中的 sep='\t' :

dataset = pd.read_csv(inputFileName, sep='\t')