numpy把多个列写入csv
时间: 2023-10-22 09:06:45 浏览: 153
pandas将numpy数组写入到csv的实例
可以使用numpy.savetxt()函数将多个列写入csv文件。以下是一个例子:
```python
import numpy as np
# 创建包含多个列的数组
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 将数据写入csv文件
np.savetxt('data.csv', data, delimiter=',')
```
这将创建一个名为data.csv的文件,其中包含以下内容:
```
1,2,3
4,5,6
7,8,9
```
其中每一行都是一个数组元素,每个元素按照逗号分隔。如果想要指定列名,可以使用numpy.savetxt()函数的header参数,例如:
```python
np.savetxt('data.csv', data, delimiter=',', header='col1,col2,col3')
```
这将创建一个包含列名的csv文件,如下所示:
```
# col1,col2,col3
1,2,3
4,5,6
7,8,9
```
阅读全文