如何從字符串塊創建 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)