要使用NumPy讀取CSV文件中的指定列,可以使用numpy.genfromtxt()
函數。以下是一個示例代碼:
import numpy as np
# 讀取CSV文件
data = np.genfromtxt('file.csv', delimiter=',')
# 選擇指定列
selected_cols = data[:, [0, 2, 4]] # 這里選擇第1、第3、第5列
print(selected_cols)
在這個示例中,我們使用genfromtxt()
函數讀取了名為file.csv
的CSV文件。默認情況下,genfromtxt()
函數會將CSV文件的每一行轉換為NumPy數組中的一行。然后,我們使用索引操作選擇了第1、第3和第5列(索引從0開始)。
如果你想指定列的名稱而不是索引,可以使用names=True
參數,然后使用列名稱進行選擇。例如:
data = np.genfromtxt('file.csv', delimiter=',', names=True)
selected_cols = data[['col1', 'col3', 'col5']]
在這個示例中,CSV文件的第一行被解析為列名稱。通過將names=True
傳遞給genfromtxt()
函數,我們告訴它將第一行視為列名稱。然后,我們使用列名稱進行選擇。