numpy repeat

最新推荐文章于 2024-04-23 04:45:00 发布

DeniuHe

最新推荐文章于 2024-04-23 04:45:00 发布

阅读量947

点赞数

分类专栏：算法文章标签：矩阵

本文链接：https://blog.csdn.net/DeniuHe/article/details/121320304

版权

算法专栏收录该内容

193 篇文章 2 订阅

订阅专栏

import numpy as np
from scipy.spatial.distance import pdist, squareform

MM = np.array([[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15],[16,17,18,19,20],[21,22,23,24,25]])
print(MM)


A = [0,0,0,1,1,1]
B = [2,2,2,3,3,3,4,4,4]



Matrix = np.ones((len(A),len(B)))
for i in range(len(A)):
    for j in range(len(B)):
        Matrix[i,j] = MM[A[i],B[j]]

print(Matrix)
A = [0,1]
B = [2,3,4]
matrix = np.ones((len(A), len(B)))
for i in range(len(A)):
    for j in range(len(B)):
        matrix[i,j] = MM[A[i],B[j]]
print(matrix)
matrix = matrix.repeat(3,axis=0).repeat(3,axis=1)

print(matrix)

[[ 1 2 3 4 5]
[ 6 7 8 9 10]
[11 12 13 14 15]
[16 17 18 19 20]
[21 22 23 24 25]]
[[ 3. 3. 3. 4. 4. 4. 5. 5. 5.]
[ 3. 3. 3. 4. 4. 4. 5. 5. 5.]
[ 3. 3. 3. 4. 4. 4. 5. 5. 5.]
[ 8. 8. 8. 9. 9. 9. 10. 10. 10.]
[ 8. 8. 8. 9. 9. 9. 10. 10. 10.]
[ 8. 8. 8. 9. 9. 9. 10. 10. 10.]]
[[ 3. 4. 5.]
[ 8. 9. 10.]]
[[ 3. 3. 3. 4. 4. 4. 5. 5. 5.]
[ 3. 3. 3. 4. 4. 4. 5. 5. 5.]
[ 3. 3. 3. 4. 4. 4. 5. 5. 5.]
[ 8. 8. 8. 9. 9. 9. 10. 10. 10.]
[ 8. 8. 8. 9. 9. 9. 10. 10. 10.]
[ 8. 8. 8. 9. 9. 9. 10. 10. 10.]]