面试题答案
一键面试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
来计算每一列的和。