Добавьте одну точку к каждой точке в массиве numpy - PullRequest
0 голосов
/ 19 марта 2020

Сб. У меня есть следующий массив numpy:

arr = numpy.array([[0,0], [1, 0], [2, 0], [3, 0]])

Как добавить один подмассив в каждый из шести подмассивов? (Скажем, если вы хотите добавить [2,1] к каждому из них, тогда результат должен быть [[2,1], [3, 1], [4, 1], [5, 1]])

Я знаю, если это одномерный массив, вы можете просто написать что-то вроде arr + 1, и он добавит 1 к каждый элемент в обр, но что в этом случае? Мне еще не удалось найти относительную информацию в документации

1 Ответ

0 голосов
/ 19 марта 2020
arr = np.array([np.append(item, [2,1]) for item in arr])

Это должно дать вам результат

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...