python二維數(shù)組怎么轉(zhuǎn)化為一維數(shù)組

小億
244
2023-09-27 03:54:26
欄目: 編程語言

要將一個(gè)二維數(shù)組轉(zhuǎn)化為一維數(shù)組，可以使用列表推導(dǎo)式（list comprehension）或者使用numpy庫(kù)中的flatten()函數(shù)。

方法1: 使用列表推導(dǎo)式

# 示例二維數(shù)組
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
# 使用列表推導(dǎo)式將二維數(shù)組轉(zhuǎn)化為一維數(shù)組
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庫(kù)的flatten()函數(shù)

import numpy as np
# 示例二維數(shù)組
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 使用flatten()函數(shù)將二維數(shù)組轉(zhuǎn)化為一維數(shù)組
new_arr = arr.flatten()
print(new_arr)  # 輸出: [1 2 3 4 5 6 7 8 9]

注意：如果使用numpy庫(kù)的flatten()函數(shù)，需要先安裝numpy庫(kù)（可以通過pip install numpy命令進(jìn)行安裝）。

0 踩