У меня есть массив размером (100, 50) . Мне нужно сгенерировать выходной массив, который представляет собой декартово произведение строк входного массива.
Для упрощения, давайте иметь входной массив:
array([[2, 6, 5],
[7, 3, 6]])
В качестве вывода я хотел бы иметь:
array([[2, 7],
[2, 3],
[2, 6],
[6, 7],
[6, 3],
[6, 6],
[5, 7],
[5, 3],
[5, 6]])
Примечание. itertools.product здесь не работает из-за размера входного вектора. Также все другие аналогичные ответы, предполагает количество строк меньше 32, что здесь не так
2 ответа
Этот вопрос задавался много раз, например здесь.
Массив размером (100, 50) слишком велик и не может быть обработан numpy. Тем не менее, меньший размер массива может быть решен. В любом случае, я предпочитаю использовать itertools для такого рода вещей:
import itertools
a = np.array([[2, 6, 5], [7, 3, 6]])
np.array(list(itertools.product(*a)))
array([[2, 7],
[2, 3],
[2, 6],
[6, 7],
[6, 3],
[6, 6],
[5, 7],
[5, 3],
[5, 6]])
a = np.array([[2, 6, 5],[7, 3, 6]])
out = np.array(np.meshgrid(a[0], a[1])).T.reshape(-1,2)
print(out)
"""
prints
[[2 7]
[2 3]
[2 6]
[6 7]
[6 3]
[6 6]
[5 7]
[5 3]
[5 6]]
"""
Похожие вопросы
Новые вопросы
python
Python - это мультипарадигмальный динамически типизированный многоцелевой язык программирования, разработанный для быстрого (для изучения, использования и понимания) и для обеспечения чистого и унифицированного синтаксиса. Обычно используются две похожие, но несовместимые версии Python, Python 2.7 и 3.x. Для специфичных для версии вопросов Python добавьте тег [python-2.7] или [python-3.x]. При использовании варианта или библиотеки Python (например, Jython, PyPy, Pandas, Numpy), пожалуйста, включите их в теги.