我已经定义了一个函数,它需要三个条目、2 个数字和一个数组。该阵列是一个 800 行 10 列的阵列。
数组中的第一列有一个州的编号(target_province)
我想要该函数做的是逐行并将每个省的列数据分配给一个空数组并返回它。
但是,似乎 if 语句被跳过了。我使用了错误的语法吗?
def get_Province_Data(target_province,data,array):
Date=[]
Data_Set=[]
f=0
for rows in array:
province=array[f][0]
if province == target_province:
np.append(Date,array[f][data])
np.append(Data_Set,array[f][3]) #gets the date data from rom f column 3
print(1)
f=f+1
return Date,Data_Set
alpha, cases=get_Province_Data(35,4,CSV_Array)
print(cases)
'''
回答1
从文件中读取的 value 默认为 String 类型。您正在比较让我们说字符串 '1' 和 int 1。
改用这个
if int(province) == target_province: