python实现txt转np,其中txt文件是float,要求转为2阶的np
时间: 2024-09-20 13:14:31 浏览: 54
python将txt文件读入为np.array的方法
在Python中,你可以使用numpy库将文本文件转换为二维数组(2D array),假设这个文本文件每一行代表一行数据,每个数值之间由空格分隔。以下是一个基本步骤:
```python
import numpy as np
# 假设文件名为"data.txt"
with open('data.txt', 'r') as file:
# 读取所有行并删除末尾的换行符
data = [line.strip() for line in file]
# 将每行数据分割成浮点数列表
data_floats = [list(map(float, row.split())) for row in data]
# 转换为2D numpy数组
data_array_2d = np.array(data_floats)
# 确保数据已经按照预期转换
assert data_array_2d.shape[1] == 2, "文件中的一行数据没有包含两个元素"
# 打印数组形状确认结果
print("转换后的二维数组形状:", data_array_2d.shape)
```
在这个示例中,`split()`函数用于根据空格分隔每个值,然后`map(float, ...)`用于将字符串转换为浮点数。最后,我们将所有这样的列表转换为一个numpy数组。
阅读全文