python读取csv文件某一列并存储为array

可以使用Python内置的csv模块来读取csv文件，并使用numpy库中的array来存储数据。下面是一个示例代码：

import csv
import numpy as np

filename = 'data.csv'  # csv文件路径
col_num = 2  # 要读取的列号，这里是第2列

# 读取csv文件并提取指定列的数据
data = []
with open(filename, 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        data.append(row[col_num])

# 将数据转换为numpy array
arr = np.array(data)

# 打印array
print(arr)

在上面的代码中，我们首先定义了csv文件的路径和要读取的列号。然后使用csv模块读取csv文件，并将指定列的数据存储到一个列表中。最后，使用numpy库中的array将列表转换为array，并进行打印。