Pythonの二次元配列を一次元配列に変換する方法

2次元配列を1次元配列に変換するには、リスト内包表記か、numpyライブラリのflatten()関数を使用します。

メソッド 1: リスト内包表記の使用

# 示例二维数组
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
# 使用列表推导式将二维数组转化为一维数组
new_arr = [element for sublist in arr for element in sublist]
print(new_arr)  # 输出: [1, 2, 3, 4, 5, 6, 7, 8, 9]

方法2: NumPy ライブラリの flatten() 関数を利用する

import numpy as np
# 示例二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 使用flatten()函数将二维数组转化为一维数组
new_arr = arr.flatten()
print(new_arr)  # 输出: [1 2 3 4 5 6 7 8 9]

NumPyライブラリのflatten()関数を用いる場合は、NumPyライブラリのインストールが必要(pip install numpyコマンドでインストール可能)。

bannerAds