列表取值的问题?

node_types=np.array([4,4,4])

adj_offset = np.zeros((node_types.shape[0], node_types.shape[0]), dtype=np.float32)

print(adj_offset)

'''
[[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]

'''
u=np.array([[1,2],[1,2],[1,0],[2,0],[2,0],[2,0]])

adj_offset[u[:, 0], u[:, 1]] = 1

'''u[:, 0]   1,1,1,2,2,2

    u[:, 1]   2,2,0,0,0,0

           [1,2] [1,2],[1,0],[2,0],[2,0],[2,0位置为1'''



print(adj_offset)
''''[[0. 0. 0.]
 [1. 0. 1.]
 [1. 0. 0.]]'''

你可能感兴趣的:(numpy,python)