MST
星途 面试题库

面试题:Python中NumPy多维数组的基本操作

使用Python的NumPy库创建一个3x3的二维数组,数组元素从1到9按顺序填充。然后计算该二维数组每一列的和,并返回结果。
47.0万 热度难度
编程语言Python

知识考点

AI 面试

面试题答案

一键面试
import numpy as np

# 创建3x3的二维数组,元素从1到9按顺序填充
arr = np.arange(1, 10).reshape(3, 3)

# 计算每一列的和
col_sum = np.sum(arr, axis=0)

print(col_sum)

上述代码首先使用np.arange(1, 10)生成从1到9的一维数组,然后通过reshape(3, 3)将其转换为3x3的二维数组。接着使用np.sum函数并指定axis=0来计算每一列的和。