通过列表理解,我能够采用字符串格式的 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/