假设我有一个如下的numpy数组
a = np.asarray([[1,2,3],[1,4,3],[2,5,4],[2,7,5]])
array([[1, 2, 3],
[1, 4, 3],
[2, 5, 4],
[2, 7, 5]])
如何为第1列中的每个唯一元素展平第2列和第3列,如下所示:
array([[1, 2, 3, 4, 3],
[2, 5, 4, 7, 5],])
谢谢您的帮助。
使用列表理解的另一种选择:
np.array([np.insert(a[a[:,0] == k, 1:].flatten(), 0, k) for k in np.unique(a[:,0])])
# array([[1, 2, 3, 4, 3],
# [2, 5, 4, 7, 5]])
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句