Python将csv某一列存为数组

要将CSV文件的某一列存储为Python数组，可以使用Python的标准库csv来读取CSV文件，并将所需列中的数据存储在数组中。下面是一个简单的示例代码来实现这个功能：

假设CSV文件（例如：data.csv）如下所示：

Name,Age,Location
John,25,New York
Alice,30,San Francisco
Bob,22,Los Angeles

现在将Age列存储为一个Python数组：

import csv

def get_column_as_array(csv_file, column_name):
    column_array = []
    with open(csv_file, newline='') as csvfile:
        reader = csv.DictReader(csvfile)
        for row in reader:
            column_array.append(row[column_name])
    return column_array

# 示例用法
csv_file = 'data.csv'
column_name = 'Age'
age_array = get_column_as_array(csv_file, column_name)
print(age_array)

运行以上代码，输出结果会是：

['25', '30', '22']

请注意，CSV文件中读取的列数据默认会以字符串形式存储在数组中。如果需要将数据转换为其他类型（例如整数或浮点数），可以在读取时进行转换，例如：

def get_column_as_array(csv_file, column_name, data_type=int):
    column_array = []
    with open(csv_file, newline='') as csvfile:
        reader = csv.DictReader(csvfile)
        for row in reader:
            column_array.append(data_type(row[column_name]))
    return column_array

# 示例用法
csv_file = 'data.csv'
column_name = 'Age'
age_array = get_column_as_array(csv_file, column_name, data_type=int)
print(age_array)

运行以上代码，输出结果将会是：

[25, 30, 22]

现在，已经成功将CSV文件中的某一列存储为Python数组，并且可以根据需要选择是否将数据转换为不同的数据类型。