如何在python variab中保存列

2024-03-28 15:56:34 发布

您现在位置:Python中文网/ 问答频道 /正文

我刚刚开始使用python,并且学习了np数组。我要做的是,将np列保存为常量值。示例:

import numpy as np

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

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

从目前的情况来看,我在终端上测试时上面的代码是不正确的。但是有没有一种方法可以将常量设置为列,或者手动测试if State = [:, 4]

TL;DR有没有办法将[:,4]保存为常量变量


Tags: importnumpy网格示例ifas地方np
1条回答
网友
1楼 · 发布于 2024-03-28 15:56:34

请考虑下面的示例代码来开始。你知道吗

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

相关问题 更多 >