我可以用numpy传递一个矩阵作为命令行输入吗？

Question

我正在做一个简单的程序，让我进入康威人生游戏的下一步，使用numpy。 现在，我能做的就是改变程序中的1和0：

seed = np.array([[0,0,0,0],[0,1,1,0],[0,1,1,1],[0,0,0,0]])  
print(nextStep(seed))

但我希望能够在命令行中传递初始数组。有什么办法吗？在

Answer 1

如果你想写

$ python life.py 0101010,0101010,0101010

然后需要来自sys模块的argv向量，它包含命令行中的所有元素，然后是一个简单的列表理解 将第一个参数拆分为逗号，从而得到矩阵的rows（注意这些行是字符串），然后构建一个子列表，转换一行中的每个字符

^{pr2}$

为了详细说明这个过程，使用一个辅助变量来代替argv[1]

In [46]: m = '0101010,0101010,0101010'

In [47]: [r for r in m.split(',')]
Out[47]: ['0101010', '0101010', '0101010']

In [48]: [[c for c in r] for r in m.split(',')]
Out[48]: 
[['0', '1', '0', '1', '0', '1', '0'],
 ['0', '1', '0', '1', '0', '1', '0'],
 ['0', '1', '0', '1', '0', '1', '0']]

In [49]: [[int(c) for c in r] for r in m.split(',')]
Out[49]: [[0, 1, 0, 1, 0, 1, 0], [0, 1, 0, 1, 0, 1, 0], [0, 1, 0, 1, 0, 1, 0]]

In [50]: np.array([[int(c) for c in r] for r in m.split(',')])
Out[50]: 
array([[0, 1, 0, 1, 0, 1, 0],
       [0, 1, 0, 1, 0, 1, 0],
       [0, 1, 0, 1, 0, 1, 0]])

In [51]: