np.savetxt用法
时间: 2023-10-11 11:05:39 浏览: 122
`np.savetxt` 是一个 NumPy 库中的函数,用于将数组保存到文本文件中。它的基本语法如下:
```python
np.savetxt(fname, X, fmt='%.18e', delimiter=' ', newline='\n', header='', footer='', comments='# ', encoding=None)
```
其中,参数含义如下:
- `fname`: 文件名或文件名路径。如果 `fname` 是一个字符串,数据将被写入到该文件中。如果 `fname` 是一个文件描述符,则数据将被写入到该文件中。如果 `fname` 以 `'.gz'` 结尾,则该文件将被压缩。
- `X`: 要保存的数组。
- `fmt`: 写入文件的格式。默认值为 `'%.18e'`,即科学计数法。
- `delimiter`: 分隔符,默认为 `' '`。
- `newline`: 行分隔符,默认为 `'\n'`。
- `header`: 文件头。
- `footer`: 文件尾。
- `comments`: 注释符号,默认为 `'# '`。
- `encoding`: 文件编码格式。
下面是一个使用 `np.savetxt` 的例子:
```python
import numpy as np
# 创建一个数组
a = np.array([[1, 2], [3, 4]])
# 保存数组到文件中
np.savetxt('data.txt', a, fmt='%d', delimiter=',')
# 从文件中读取数据
b = np.loadtxt('data.txt', delimiter=',')
print(b)
# 输出:
# [[1 2]
# [3 4]]
```
这里,我们使用 `np.savetxt` 将数组 `a` 保存到名为 `'data.txt'` 的文件中,分隔符为逗号,格式为整数。然后,我们使用 `np.loadtxt` 从该文件中读取数据,再将其存储在数组 `b` 中,并打印出来。
阅读全文