Ошибка скалярных массивов в функции диапазона - Python

Question

Я пытаюсь поместить определенные значения в одномерный массив (на основе индекса), но когда я вставляю индекс в функцию диапазона в цикле for - я получаю ошибку:

TypeError: only integer scalar arrays can be converted to a scalar index

Вот соответствующий код:

import numpy as np
from scipy.interpolate import interp1d

x = np.arange(0, 24, 0.05)   # (Start #, Last #, Jump)
x.size
n, = np.where(x == 6)
y = x

for i in range(0,n):
    y[i] = 0 

Я не понимаю, почему это происходит, потому что n должен быть сам индекс, нет?

Спасибо!

Answer 1

where возвращает ndarray. Вам нужно использовать только первый элемент

for i in range(n[0]):
    y[i] = 0