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

此代碼的問題在於它讀取整個數據文件並將其存儲為數據集。 那么如果我想單獨閱讀這 2 列，我該怎么辦

提前致謝

Answer 1

我認為您有separator tab ，因此需要將read_csv默認sep=',' （id 無sep參數）更改為sep='\\t' ：

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