如何在python变量中保存列

Question

我刚开始使用python，我已经了解了np数组。我想要做的是，将np列保存为常量值。例如：

import numpy as np

SEA = np.array([:,4])

我正在创建一个5x5的网格世界，我想要一个值，我可以检查如果玩家落在第4列的任何地方，他们就会淹死并失败，所以我可以检查if State == SEA: break或类似的东西。

目前，我在终端测试时上面的代码不正确。但有没有办法将常量设置为列或我手动测试if State = [:, 4]

TL; DR有一种方法可以将[：，4]保存为const变量

Answer 1

考虑这个示例代码，以帮助您入门。

import numpy as np

SEA_COLUMN = 3
USER_POSITION_VALUE = 1

def is_user_in_sea(list_array):
    if USER_POSITION_VALUE in list_array[:, SEA_COLUMN]:
        print('In Sea')
    else:
        print('Safe')

list_array = np.zeros((5, 5), dtype = np.int32)
list_array[1][2] = USER_POSITION_VALUE
is_user_in_sea(list_array)
# Safe

list_array[4][SEA_COLUMN] = USER_POSITION_VALUE   # Not resetting the old value
is_user_in_sea(list_array)
# In Sea