L oop до заполнения матрицы? - PullRequest
0 голосов
/ 29 мая 2020

У меня есть условный оператор, который добавляет строку двоичных значений из матрицы A в матрицу B. Я хочу поместить это в al oop, чтобы он продолжал добавлять строки из матрицы A, пока матрица B не будет заполнена. В настоящее время матрица B инициализирована как матрица нулей 10 на 10. Нужно ли мне инициализировать матрицу B по-другому, чтобы создать это условие, или есть способ сделать это как есть?

Ниже примерно так выглядит мой код на данный момент

from random import sample
import numpy as np

matrixA = np.random.randint(2, size=(10,10))

matrixB = np.zeros((10,10))

x, y = sample(range(1, 10), k=2)

if someCondition:
    matrixB = np.append(matrixB, [matrixA[x]], axis=0)
else:
    matrixB = np.append(matrixB, [matrixA[y]], axis=0)

1 Ответ

1 голос
/ 29 мая 2020

вам не нужно для этого al oop. Это действительно легко сделать с помощью интеллектуальной индексации. Например:

import numpy as np

A = np.random.randint(0, 10, size=(20,10))
B = np.empty((10, 10))
print(A)
# Copy till the row that satisfies your conditions. Here I assume it to be 10
B = A[:10, :]
print(B)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...