python 一维数组对齐并填充0

import numpy as np

def arr_pad_align(arr1, arr2):
    arr1_len = len(arr1)  # 取第一个数组的长度
    arr2_len = len(arr2)  # 取第二个数组的长度
    arr_list = [(arr1), (arr2)]  # 将数组放入list容器
    arr_matrix = [[arr1_len, arr2_len]]  # 将输入的长度放入二维数组，方便取索引
    max_index = np.argmax(arr_matrix, 1).squeeze(0)  # 取最大长度数组的索引
    min_index = np.argmin(arr_matrix, 1).squeeze(0)  # 取最小长度数组的索引
    pad_len = np.max([arr1_len, arr2_len]) - np.min([arr1_len, arr2_len])  # 拿到填充的长度
    arr_min = arr_list[min_index]  # 拿到填充数组
    pad_arr = np.pad(arr_min, (0, pad_len), 'constant', constant_values=0)  # 进行填充，尾部为0
                            # (0, pad_len), 前面填充0个0，后面填充pad_len个0
    return pad_arr, arr_list[max_index]  # 返回填充后的结果

arr1 = np.array([1, 2, 3, 4])
arr2 = np.array([1, 2, 3, 4, 5])
print(arr1)
print(arr2)

arr1, arr2 = arr_pad_align(arr1, arr2)
print(arr1)
print(arr2)

# 结果
[1 2 3 4]
[1 2 3 4 5]
[1 2 3 4 0]
[1 2 3 4 5]