import numpy as np
att = [[0, 0, 0, 1, 1, 0, 0, 1],[0, 0, 0, 0, 0, 0, 0, 0]]
att = np.array(att)
trainval = [0, 1]
# astype('float32')==1表示数值为1的时候记为1,不为1的时候记为0
# 对array数据所有索引取平均
weight_trainval = np.mean(att[trainval, :].astype('float32')==1, axis=0).tolist()
print("weight_trainval===========",weight_trainval)
04-09
9379