numpy array存csv
时间: 2023-07-28 17:07:24 浏览: 43
可以使用 `numpy.savetxt` 函数将 NumPy 数组保存为 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.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
```
默认情况下, `np.savetxt` 将浮点数保存为科学计数法形式,可以使用 `fmt` 参数指定输出格式。例如,将 `fmt` 设置为 `"%.2f"` 将保留两位小数:
```python
import numpy as np
# 创建一个示例数组
data = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
# 将数组保存到 CSV 文件,保留两位小数
np.savetxt("data.csv", data, delimiter=",", fmt="%.2f")
```
这个代码将创建一个名为 `data.csv` 的文件,并将数组保存到该文件中。在这个示例中,文件内容如下:
```
1.00,2.00,3.00
4.00,5.00,6.00
7.00,8.00,9.00
```
如果数组中的每个元素不是浮点数,则可以使用 `fmt` 参数指定其他格式,例如 `%d` 表示整数。