利用numpy提取csv中的特定列并另存为新的csv 示例代码

假设原始的csv文件名为"original.csv"，包含4列数据，分别为姓名、性别、年龄、分数。现在我们需要提取其中的姓名和分数两列并另存为新的csv文件"new.csv"，可以使用以下代码：

import numpy as np

# 读取原始csv文件
data = np.genfromtxt('original.csv', delimiter=',', dtype=str)

# 提取姓名和分数两列
new_data = data[:, [0, 3]]

# 将提取的数据保存为新的csv文件
np.savetxt('new.csv', new_data, delimiter=',', fmt='%s')

解释：

1. 首先使用numpy的genfromtxt函数读取原始的csv文件，并将数据存储在一个numpy数组中。其中，delimiter参数指定了csv文件中的分隔符，dtype参数指定了数据类型为字符串。

2. 然后使用numpy数组的切片功能提取姓名和分数两列数据，这里使用[:, [0, 3]]表示提取所有行的第0列和第3列数据。

3. 最后使用numpy的savetxt函数将提取的数据保存为新的csv文件。其中，delimiter参数指定了保存时的分隔符，fmt参数指定了保存的数据格式为字符串。