Извлечение данных из двумерных массивов NumPy
Если двумерный массив numpy имеет регулярную структуру, то есть каждая строка и каждый столбец содержат фиксированное количество элементов, работа с подмножествами становится очень простой. Рассмотрите код ниже, в котором элементы "a" и "c" извлекаются из списка списков.
# numpy
import numpy as np
np_x = np.array(x)
np_x[:, 0]
Индексы до запятой относятся к строкам, а после запятой — к столбцам. Символ : используется для срезов; в данном примере он указывает Python включить все строки.
Это упражнение является частью курса
Введение в Python
Инструкции к упражнению
- Выведите на экран 50-ю строку массива
np_baseball. - Создайте новую переменную
np_weight_lb, содержащую весь второй столбец массиваnp_baseball. - Выберите рост (первый столбец) 124-го бейсболиста из массива
np_baseballи выведите его на экран.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
import numpy as np
np_baseball = np.array(baseball)
# Print out the 50th row of np_baseball
# Select the entire second column of np_baseball: np_weight_lb
# Print out height of 124th player