如何在 Python 中将数组的某些元素乘以不同的 sclar

Question

例如，如果我有一个 np.ones(N) 数组，则该数组将垂直成形。

接下来，我会做 对于范围内的 i(1,int(N/2))： （这是我想将数组的一半乘以某个常数的地方） 对于范围内的 i(int(N/2),int(N))： （这是我想将数组的另一半乘以不同的常数的地方）

我该怎么做？

Answer 1

使用数组索引

示例

import numpy as np

N = 6
arr = np.ones(N)   # All ones

arr[:N//2] *= 2    # Multiply first half by 2
arr[N//2:] *= 4    # Multiply second half by 4

print(arr):        # Out: [2. 2. 2. 4. 4. 4.]