如何从字符串块创建 2D numpy 数组

Question

通过列表理解，我可以获取 20x20 字符串格式的数字块，并将其转换为整数列表列表。 数字由空格分隔，行由换行符分隔。

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

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

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

有没有办法在不使用创建中间 python 列表的情况下完成这项任务？

Answer 1

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

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

---

或者作为更通用的解决方案，遵循@mihammad 的解决方案：

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

Answer 2

对于任何 2D 网格更通用：

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