numpy array 提取其中每个array中的第i个元素

1.最初数据格式

list1=np.array([1,2,3,1,1])
list2=np.array([4,5,6,2,3])
list3=np.array([7,8,9,5,6])
list4=np.array([17,18,19,6,7])
list5=np.array([17,18,19,4,5])

list = [
    list1,
    list2,
    list3,
    list4,
    list5
]
arr=np.array(list)

• 数据格式 (两层数组)

[[ 1  2  3  1  1]
[ 4  5  6  2  3]
[ 7  8  9  5  6]
[17 18 19  6  7]
[17 18 19  4  5]]

2.需要转成的格式

[ 1  4  7 17 17]
[ 2  5  8 18 18]
[ 3  6  9 19 19]
[1 2 5 6 4]
[1 3 6 7 5]

即分别取出原数组第一层的每个i个元素,并放在相应的数组中

3.code

思路:先把array数组转为list,在进行分离和append操作,最后转成数组

def extractTimeImage(list_arr, list1_arr, list2_arr, list3_arr, list4_arr, list5_arr, arr):
    for i in range(len(arr)):
        list_arr.append((arr[i]).tolist())
        list1_arr.append(list_arr[i][0])
        list2_arr.append(list_arr[i][1])
        list3_arr.append(list_arr[i][2])
        list4_arr.append(list_arr[i][3])
        list5_arr.append(list_arr[i][4])

    list1_arr = np.array(list1_arr)
    list2_arr = np.array(list2_arr)
    list3_arr = np.array(list3_arr)
    list4_arr = np.array(list4_arr)
    list5_arr = np.array(list5_arr)
    return list1_arr, list2_arr, list3_arr, list4_arr, list5_arr


list_arr=[]
list1_arr=[]
list2_arr=[]
list3_arr=[]
list4_arr=[]
list5_arr=[]
new_arr1,new_arr2,new_arr3,new_arr4,new_arr5=extractTimeImage(list_arr, list1_arr, list2_arr, list3_arr, list4_arr, list5_arr, arr)

• output

[ 1  4  7 17 17]
[ 2  5  8 18 18]
[ 3  6  9 19 19]
[1 2 5 6 4]
[1 3 6 7 5]