gpt4 book ai didi

python - numpy 数组子集的元素平方

转载 作者:行者123 更新时间:2023-12-02 01:40:37 31 4
gpt4 key购买 nike

我有以下 numpy 数组

np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])

我只想通过按原样保留第一列来计算第二列和第三列的元素平方,并产生结果,

np.array([[1,  4,   9],
[4, 25, 36]],
[7, 64, 81]])

我尝试过什么我提取了第一列。然后我提取了第二列和第三列。使用 numpy.square 函数找到正方形。

arr1 = arr[:, 0]
arr2 = np.square(arr[:, 1:])

然后将它们连接起来

np.c_[arr1, arr2]

有单步解决方案吗?

最佳答案

使用[:, 1:]选择非第一列的所有元素。

arr[:, 1:] **= 2

关于python - numpy 数组子集的元素平方,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/71626166/

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