#python #numpy
Вопрос:
У меня есть массив numpy, сгенерированный из PLSRegression с количеством компонентов, равным 2
(array([[ 0.48362376, 1.15908429], [-2.47132438, 0.44989534], [-2.60893728, 0.5381059 ], [-2.53155766, 0.3469423 ], [-1.89554596, 0.73211665]]))
Я хочу получить координаты X_1 и X_2 для построения 2-мерной диаграммы рассеяния. Таким образом, в этом случае координаты X_1 и X_2 будут
X_1 = [0.48362376,-2.47132438,-2.60893728,-2.53155766,-1.89554596] X_2 = [1.15908429, 0.44989534, 0.5381059,0.3469423,0.73211665]
Ответ №1:
Пусть ваш массив будет x
. Затем вы можете использовать индексацию следующим образом:
x_1 = x[:, 0] x_2 = x[:, 1]
x[:, 0]
это означает, что вы берете каждую строку ( :
) и первый столбец ( 0
). Таким образом, вы получаете массив с элементами первого столбца.
Помните, что с x[1, 0]
помощью вы берете элемент из второй строки и первого столбца. Вместо того, чтобы использовать двоеточие вместо определенного числа, вы просто выбираете «все» элементы в соответствующем измерении.