python - 从字典数组中提取数字数组

Question

我有以下 2x2 字典数组:

a = np.array([[{'x_id':0, 'y_id':0},{'x_id':1, 'y_id':1}],[{'x_id':2, 'y_id':0},{'x_id':3, 'y_id':1}]])

我想获取与键 'x_id'、[[0, 1], [2, 3]] 的值相对应的 2x2 数字数组，即:

0  1
2  3

除了双for循环之外还有其他方法吗？即:

numbers = [[a[i,j]['x_id'] for j in range(2)] for i in range(2)]

Answer 1

如果您在代码中谈论显式 for 循环，则可以展平您的数组并使用单个 for 循环完成工作，然后 reshape 你的最终数组

numbers = np.array([i['x_id'] for i in a.flatten()]).reshape(a.shape)
# array([[0, 1],
#        [2, 3]])

另一种解决方案是在展平数组上使用itemgetter作为

import operator
numbers = np.array(list(map(operator.itemgetter('x_id'), a.flatten()))).reshape(a.shape)

性能:两种方法花费的时间相似

%timeit np.array([i['x_id'] for i in a.flatten()]).reshape(a.shape)
# 4.16 µs ± 676 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

%timeit np.array(list(map(operator.itemgetter('x_id'), a.flatten()))).reshape(a.shape)
# 4.9 µs ± 1.26 µs per loop (mean ± std. dev. of 7 runs, 100000 loops each)