Python - 从多个列集轻松导入csv数组

Question

我正在尝试通过csv导入此数据集。

import numpy as np
import csv 

Snap1 = []
Snap2 = []

with open('BrownM.csv', 'r') as f:
    reader = csv.reader(f, delimiter=',')
    for row in reader:
        Snap1.append(row[0])
        Snap2.append(row[1])

Snap1 = np.array(Snap1)

这样可以正常工作，但尝试追加89列并将其转换为数组非常耗时。

他们的疯狂方法是否更容易？

Answer 1

您可以尝试使用numpy中的genfromtxt功能。通过将分隔符设为,，它会将csv文件加载到可以切割成列的2D数组中。

像这样：

import numpy as np

with open('BrownM.csv', 'r') as f:
    data = np.genfromtxt(f, delimiter=',')
Snap1 = data[:, 0]
Snap2 = data[:, 1]