python numpy存储为csv 代码
时间: 2023-11-12 13:18:57 浏览: 92
下面是使用NumPy将数组存储为CSV文件的示例代码:
```python
import numpy as np
# 创建一个numpy数组
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 将数组保存为CSV文件
np.savetxt('data.csv', data, delimiter=',')
```
在上面的代码中,我们首先使用NumPy创建一个包含数据的二维数组。然后,我们使用`np.savetxt`函数将数组保存为CSV文件。`np.savetxt`函数有三个参数:
- `fname`: 保存CSV文件的文件名或文件路径。
- `X`: 要保存的数组。
- `delimiter`: CSV文件中的分隔符,默认为逗号。在上面的示例中,我们使用逗号作为分隔符。
运行上面的代码后,将创建一个名为`data.csv`的文件,其中包含以下内容:
```
1.000000000000000000e+00,2.000000000000000000e+00,3.000000000000000000e+00
4.000000000000000000e+00,5.000000000000000000e+00,6.000000000000000000e+00
7.000000000000000000e+00,8.000000000000000000e+00,9.000000000000000000e+00
```
注意,保存的CSV文件包含科学计数法的数字。如果需要,可以使用NumPy的`set_printoptions`函数设置打印选项,例如:
```python
np.set_printoptions(suppress=True)
```
这将禁用科学计数法,并将数字打印为常规浮点数。
阅读全文