gpt4 book ai didi

python - 矩阵python的向量(positionx,positiony,value)

转载 作者:行者123 更新时间:2023-11-28 22:52:52 26 4
gpt4 key购买 nike

我有一个值矩阵,我想要一个包含矩阵位置和值的向量。

第一个想法:double-bucle 并存储索引和值

第二个想法:创建一个带有序列的 ndarray

还有其他最有效的想法吗?

对不起我的英语。

输入(图像 1C):

[[0 1 2 3 4 5]
[6 7 8 9 10 11]]

输出:

[[0 0 1][0 1 2] [0 2 3] [0 3 4] [0 4 5] [1 0 6] [1 1 7] [1 2 8] [1 3 9] [1 4 10] [1 5 11] ... ]

谢谢大家的回答,你们帮助解决了问题,理解了numpy。

最佳答案

这是一个非常简洁的方法:

import numpy as np
arr = np.random.random((2, 6))
x, y = np.indices(arr.shape)
output = np.column_stack([x.ravel(), y.ravel(), arr.ravel()])

这里有一个非常可爱的方法可以最大限度地减少内存使用:

import numpy as np
arr = np.random.random((2, 6))
output = np.indices(arr.shape + (1,), dtype=arr.dtype)
output[-1, ..., 0] = arr
output = output.reshape(arr.ndim + 1, -1).T

关于python - 矩阵python的向量(positionx,positiony,value),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/20024214/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com