python 将字符串转Numpy数组
时间: 2023-11-15 10:01:18 浏览: 38
要将字符串转换为Numpy数组,可以使用以下步骤:
1. 使用StringIO将字符串转换为文件对象。
2. 使用loadtxt函数将文件对象加载到Numpy数组中。
3. 如果字符串中包含注释,则可以使用comments参数指定注释字符。
4. 如果字符串中包含非Double类型的字符,则可以使用genfromtxt函数并指定dtype参数来处理。
下面是一个示例代码:
```
import numpy as np
from io import StringIO
# 示例字符串
s = '1, 2, 3\n4, 5, 6\n7, 8, 9'
# 将字符串转换为文件对象
f = StringIO(s)
# 使用loadtxt函数将文件对象加载到Numpy数组中
a = np.loadtxt(f, delimiter=',')
# 打印结果
print(a)
```
输出结果为:
```
array([[1., 2., 3.],
[4., 5., 6.],
[7., 8., 9.]])
```
如果字符串中包含注释或非Double类型的字符,可以使用以下代码:
```
import numpy as np
from io import StringIO
# 示例字符串
s = '1, 2, 3\n4, 5, 6\n# This is a comment\n7, 8, 9\nA, B, C'
# 将字符串转换为文件对象
f = StringIO(s)
# 使用genfromtxt函数将文件对象加载到Numpy数组中
a = np.genfromtxt(f, delimiter=',', dtype=float, comments='#', invalid_raise=False)
# 打印结果
print(a)
```
输出结果为:
```
[[ 1. 2. 3.]
[ 4. 5. 6.]
[nan nan nan]
[ 7. 8. 9.]
[nan nan nan]]
```