对于两个数组a和b,
a = np.array([[1],[2],[3],[4]])
b = np.array(['a', 'b', 'c', 'd'])
我想生成以下数组
c = np.array([[1, 'a'], [2, 'b'], [3, 'c'], [4, 'd']])
有办法有效地做到这一点吗?
你需要:
import numpy as np
a = np.array([[1],[2],[3],[4]])
b = np.array(['a', 'b', 'c', 'd'])
print(np.array(list(zip(np.concatenate(a), b))))
输出:
[[1, 'a'], [2, 'b'], [3, 'c'], [4, 'd']]
替代解决方案
print(np.stack((np.concatenate(a), b), axis=1))
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句