Доступ к элементу ndarray с shape = (dim, L, 2) - PullRequest
2 голосов
/ 21 октября 2019

Сейчас я работаю с numpy.ndarray. Я инициализирую многомерную матрицу:

phi = np.zeros(dim_tuple(dim,L)+(2,))

, где dim_tuple (dim, L) возвращает кортеж (L, L, ..., L) с диммерным временем L.

Я хотел бы создать функцию, принимающую в качестве аргументов:

  • массив x длиной dim
  • число N: 0 или 1

, который обновит значение элемента [x [0], x [1], ..., x [dim], N] матрицы phi.

С этого момента я попробовал это:

index = np.append(x,N)

phi[index]=new_value

за исключением того, что он не работает. На самом деле, я проверяю, что она возвращает эту команду:

phi[[x[0],x[1],...,x[dim],N]] = new_value

Я попытался найти ответ в Интернете, но мне не удалось.

Спасибо заВаша помощь!

Жереми

1 Ответ

1 голос
/ 21 октября 2019

Я думаю, что вы хотите:

phi[tuple(x) + (N,)] = new_value
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...