python数组增加索引_向numpy数组追加索引

以下是代码的修改版本，其中包含一些建议和注释：import numpy as np

# when indices are 0, 1, 2, ... don't use dictionary

# also for integer values use randint

enc_ = np.random.randint(1, 24, (24,))

ret_ = np.random.randint(1, 24, (24,))

data = np.random.rand(24,24)

# np.corrcoef is vectorized, no need to loop:

corr_matrix = np.corrcoef(data)

# the following is the clearest, but maybe not the fastest way of generating

# your index array:

ind_matrix = np.argwhere(np.equal.outer(enc_, ret_))

# this can't be used for indexing directly, you'll have to choose

# one of the following idioms

# EITHER spread to two index arrays

I, J = ind_matrix.T

# or directly I, J = np.where(np.equal.outer(enc_, ret_))

# single index

print(corr_matrix[I[1], J[1]])

# multiple indices

print(corr_matrix[I[[1,2,0]], J[[1,2,0]]])

# whole row

print(corr_matrix[I[1]])

# OR use tuple conversion

ind_matrix = np.array(ind_matrix)

# single index

print(corr_matrix[(*ind_matrix[1],)])

# multiple indices

print(corr_matrix[(*zip(*ind_matrix[[1,2,0]],),)])

# whole row

print(corr_matrix[ind_matrix[1, 0]])

# OR if you do not plan to use multiple indices

as_tuple = list(map(tuple, ind_matrix))

# single index

print(corr_matrix[as_tuple[1]])

# whole row

print(corr_matrix[as_tuple[1][0]])