pythonでのnumpy配列の変換方法

Question

a = array([[1, 2, 3]
[4, 5, 6]
[7, 8, 9]])
を
a =array([[123
456
789]])
にする方法を教えて下さい
よろしくお願いします。

Comments

[can110]

提示コードはpythonとして実行できません。arrayはnp.arrayのことでしょうか？

Answer 1

import numpy as np
from functools import reduce

a = np.array([[1,2,3],[4,5,6],[7,8,9]])
ret = np.array([np.apply_along_axis(lambda i:reduce(lambda a,b:10*a+b,i), 1, a)])
print(ret) # [[123 456 789]]

Answer 2

1. astype(str) で各要素を str に変換する。
2. apply_along_axis() で行ごとにその行の要素を結合する関数 lambda x: ''.join(x) を適用する。
3. astype(int) で各要素を int に変換する。

import numpy as np

a = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])

b = np.apply_along_axis(lambda x: ''.join(x), 1, a.astype(str)).astype(int)
print(b)
# array([123, 456, 789])