gpt4 book ai didi

python - 旋转阵列的特定区域

转载 作者:行者123 更新时间:2023-12-04 01:02:35 25 4
gpt4 key购买 nike

<分区>

我有一个 NxN 矩阵,我想选择该矩阵的特定部分并将该区域顺时针旋转 90 度。我必须在这些条件下指定此旋转:

  • (a, b)位置为左上角,
  • (a + c, b + c) 位置为右下角,
  • 边长为c + 1。

所以,当我给定 a=2、b=3 和 c=3 时,输出将是这样的:

example

代码:

import numpy as np
N = int(input())
S = int(input())
arr = np.array(range(0,N*N))
arr.shape = N,N
for i in range(S):
a,b,c = [int(x) for x in input().split()]
arr[a:a+c+1,b:b+c+1] = np.rot90(arr[a:a+c+1,b:b+c+1], axes=(1,0))
print(arr);

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com