python - 如何从一个字符串 block 创建一个二维 numpy 数组

标签 python numpy

通过列表理解,我能够采用字符串格式的 20x20 数字 block ,并将其转换为整数列表列表。数字由空格分隔,行由换行符分隔。

grid = [[int(x) for x in line.split()] for line in nums.split('\n')]

但是,我想要的是使用 numpy 来提高速度。我可以将 np.asarray() 与我的中间列表一起使用,但我不认为这是对 numpy 的有效使用。

我也尝试过使用 np.fromstring(),但我无法弄清楚使其适用于二维数组的逻辑。

有没有什么方法可以在不使用创建中间 python 列表的情况下完成这个任务?

最佳答案

你可以使用 np.fromstring将空格设置为分隔符并将 reshape 设置为所需的形状:

np.fromstring(s, sep=' ').reshape(20, 20)

或者作为更通用的解决方案:

rows = s.count('\n') + 1
np.fromstring(s, sep=' ').reshape(-1, rows)

关于python - 如何从一个字符串 block 创建一个二维 numpy 数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/58307549/

相关文章:

python - 使用 Python requests 库获取带有 cookie 的页面

python - 在 Python 脚本中保护密码

python - 准确找到最小的特征值

python - 为什么我得到 "ufunc ' multiply' did not contain a loop with signature matching types dtype ('S32' ) dtype ('S32' ) dtype ('S32' )"with values from raw_input

python - 从 .ckpt 和 .meta 文件中获取输入和输出节点名称

Python - 使用格式/f 字符串输出带有 0 填充和中心的十六进制

python - numpy 数组与 nan 与标量的不等式比较

python - 将 dd、mm、yy 数组转换为 numpy 日期时间

python - 用步幅组成 numpy 数组

python - Pandas - 逗号分隔行中的每个字符串在数据框中出现的频率