利用numpy提取csv中的特定列并另存为新的csv 示例代码
时间: 2024-05-07 07:16:06 浏览: 84
假设原始的csv文件名为"original.csv",包含4列数据,分别为姓名、性别、年龄、分数。现在我们需要提取其中的姓名和分数两列并另存为新的csv文件"new.csv",可以使用以下代码:
```python
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参数指定了保存的数据格式为字符串。
相关问题
python中将numpy数据另存为csv文件
可以使用numpy库中的`numpy.savetxt()`函数将numpy数据保存为csv文件,具体操作如下:
```python
import numpy as np
# 生成一个numpy array
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 将数据保存为csv文件
np.savetxt('data.csv', data, delimiter=',')
```
其中,第一个参数为保存的文件名,第二个参数为要保存的numpy array,第三个参数为分隔符,可以根据需要进行修改。以上代码会将数据保存在当前目录下的`data.csv`文件中。
阅读全文