Python: Numpy объединяет массивы в списки 2x1

Question

Python: Numpy объединяет массивы в списки 2x1

1

Я надеюсь объединить два массива

 A: ([1,2,5,8])
 B: ([4,6,7,9])

в

 C: ([[1,4],
      [2,6],
      [5,7],
      [8,9]])

Я попытался вставить, добавить и объединить, они объединяют все элементы вместе, не давая размерности C.

Я новичок в Python, любая помощь будет оценена.

lydias 30 июль 2018, в 20:01

Источник

0

print(list(map(list, zip(A,B)))) ?
Rakesh 30 июль 2018, в 17:48

Теги:

python

arrays

numpy

list

2 ответа

1

Согласно вашему первоначальному подходу вам нужно всего лишь использовать zip, который возвращает итератор кортежей, где i-й кортеж содержит i-й элемент из каждой последовательности аргументов или итераций.

import numpy

A = numpy.array([1,2,5,8])
B = numpy.array([4,6,7,9])

print(list(zip(A, B)))

Он напечатает:

[(1, 4), (2, 6), (5, 7), (8, 9)]

Waleed Iqbal 30 июль 2018, в 16:14

Ещё вопросы

user3483203 · Accepted Answer · 2018-07-30T16-21-00.000Z

1

Лучший ответ

Используйте numpy.column_stack:

Stack 1-D массивы как столбцы в двумерный массив

np.column_stack((A, B))

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

user3483203 30 июль 2018, в 16:21

0

Как вы выбираете 1 или 2 из сложенного массива? Выбрать отдельный элемент из подсписков?
lydias 30 июль 2018, в 18:14
0

Чтобы выбрать из первого столбца, начните индексирование с помощью arr[:, 0] , а для второго столбца используйте arr[:, 1]
user3483203 30 июль 2018, в 18:15
0

Я попробовал это, но это даст мне целую колонку. Мне нужен только отдельный элемент, такой как 1,2 или 4
lydias 30 июль 2018, в 18:17
0

Просто прочитайте о индексировании NumPy, если вы хотите 1 , вы бы использовали первую строку, первый столбец, поэтому arr[0, 0]
user3483203 30 июль 2018, в 18:18

Показать ещё 2 комментария