Распакуйте кортежи в массив для визуализации графика

#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] помощью вы берете элемент из второй строки и первого столбца. Вместо того, чтобы использовать двоеточие вместо определенного числа, вы просто выбираете «все» элементы в соответствующем измерении.