numpy-基础练习

输入整数n,创建一个3*4的二维数组,数组元素从n开始,依次递增。
输出第1,2行第1,2,3列的元素

代码如下:

import numpy as np

n = int(input("请输入整数n:"))
arr = np.arange(n, n+12).reshape(3,4)

print("第1,2行第1,2,3列的元素为:")
print(arr[:2, :3])

解释一下代码:

首先从用户输入中获取整数n
使用numpy中的arange()函数,从n开始生成12个连续的整数,即从n到n+11,然后使用reshape()将它们转换成一个3行4列的二维数组
输出第1,2行第1,2,3列的元素,使用切片操作arr[:2, :3]来获取这些元素,其中[:2]表示前两行,[:3]表示前三列。