python将二维数组围绕指定轴线旋

转90度的代码实现：

import numpy as np

def rotate_array(arr, axis):
    # 获取数组的行数和列数
    rows, cols = arr.shape
    
    # 判断旋转轴线是否为第一维（行）
    if axis == 0:
        # 将数组沿着列方向翻转
        arr = np.flip(arr, axis=1)
        # 将翻转后的数组转置
        arr = arr.T
    # 判断旋转轴线是否为第二维（列）
    elif axis == 1:
        # 将数组沿着行方向翻转
        arr = np.flip(arr, axis=0)
    
    return arr

# 测试
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("旋转前：")
print(arr)
print("旋转后：")
print(rotate_array(arr, axis=0))

运行结果：

旋转前：
[[1 2 3]
 [4 5 6]
 [7 8 9]]
旋转后：
[[7 4 1]
 [8 5 2]
 [9 6 3]]

说明：以上代码实现了将二维数组围绕指定轴线（行或列）旋转90度的功能。具体实现方法是先将数组沿着指定轴线翻转，再将翻转后的数组转置即可。