从多维数组(PYTHON)返回任意行
从多维数组(PYTHON)返回任意行
提问于 2017-09-01 05:33:23
我正在使用一个5维数组，即python中的5x5x5x5。
我怎样才能使一个函数
- 将获得一个整数i (1<=i<=5)，其中一个维要迭代，另一个4维的列表( a，b，c，d)要锁定在其中。
- 返回长度为5的一维列表。
例如：
bb = (5维阵列)
rowExtract(3，2,3,0,2，2，bb)
*[ bb2i2表示I在范围(5)]返回*
3代表第3维，2,3,0,2表示其他维度。
我可以通过硬编码5种不同的场景来做到这一点，但是有更简单的方法吗？
回答 3
Stack Overflow用户
发布于 2017-09-01 07:40:02
下面是使用循环的一种方法。
def extract_row(row, arr, bb):
N = len(arr) + 1
pre_mat = bb
for i in range(row):
pre_mat = pre_mat[arr[i]]
ans = []
for i in range(N):
mat = pre_mat[i]
for j in range(row + 1, N):
mat = mat[arr[j - 1]]
ans.append(mat)
return ansStack Overflow用户
发布于 2017-09-01 05:41:39
我想没有别的办法了。所以像这样的事情
def extractRow(row, arr, bb):
if row == 0:
return [bb[i][arr[0]][arr[1]][arr[2]][arr[3]] for i in range(5)]
elif row == 1:
return [bb[arr[0]][i][arr[1]][arr[2]][arr[3]] for i in range(5)]
...Stack Overflow用户
发布于 2017-09-01 07:47:00
一种可能的解决方案是使用迭代器来生成索引序列。“缺失”维度可以插入到arr中，每次通过生成器的循环，它都会用下一个值填充该索引。每次生成生成器时，它都生成一个包含从bb提取下一个值所需的索引的5元素数组。然后您可以从bb中选择您想要的元素。
def extract_row(row, bb, arr):
retval = []
for a0, a1, a2, a3, a4 in iter_indices(row, arr):
retval.append(bb[a0][a1][a2][a3][a4])
return retval
def iter_indices(row, arr):
arr.insert(row, None)
for i in range(5):
arr[row] = i
yield arr
# TEST CODE FOLLOWS
g = 0
def deepen(b, n):
global g
if n == 1:
b.append(g)
b.append(g+1)
b.append(g+2)
b.append(g+3)
b.append(g+4)
g += 5
return b
for m in range(5):
b.append(deepen([], n-1))
return b
a = []
deepen(a, 5)
# Run it using the test matrix a
print(extract_row(1, a, [1, 2, 3, 4]))
print(a[1][0][2][3][4], a[1][1][2][3][4], a[1][2][2][3][4], a[1][3][2][3][4], a[1][4][2][3][4])
Output is:
[694, 819, 944, 1069, 1194]
694 819 944 1069 1194页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/45993827
复制相关文章
相似问题
腾讯云开发者
