Сортировка массивов — фундаментальная операция в программировании, и Python предоставляет различные методы для выполнения этой задачи. В этой статье блога мы рассмотрим несколько методов сортировки массива по определенному столбцу, используя разговорный язык и примеры кода. Независимо от того, являетесь ли вы новичком или опытным разработчиком Python, это руководство поможет вам понять различные подходы к сортировке массивов по столбцам.

Метод 1: использование встроенной функции sorted()
Функция sorted() — это универсальный инструмент для сортировки массивов в Python. Чтобы отсортировать массив по определенному столбцу, мы можем использовать параметр key и передать лямбда-функцию, которая возвращает желаемое значение столбца. Вот пример:

array = [
    [1, 4, 7],
    [2, 5, 8],
    [3, 6, 9]
]
sorted_array = sorted(array, key=lambda x: x[1])
print(sorted_array)

Выход:

[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

Метод 2: использование функцииoperator.itemgetter()
Операторный модуль Python предоставляет удобную функцию itemgetter(), которая позволяет нам извлекать определенные столбцы из вложенного массива. Мы можем объединить itemgetter() с функцией sorted() для сортировки по столбцам. Вот пример:

from operator import itemgetter
array = [
    [1, 4, 7],
    [2, 5, 8],
    [3, 6, 9]
]
sorted_array = sorted(array, key=itemgetter(1))
print(sorted_array)

Выход:

[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

Метод 3: использование numpy.argsort()
Если вы работаете с большими массивами или вам требуются сложные числовые операции, библиотека numpy может оказаться чрезвычайно полезной. Функция argsort() в numpy возвращает индексы, которые будут сортировать массив. Применяя argsort() к определенному столбцу, мы можем добиться сортировки по столбцам. Вот пример:

import numpy as np
array = np.array([
    [1, 4, 7],
    [2, 5, 8],
    [3, 6, 9]
])
sorted_indices = np.argsort(array[:, 1])
sorted_array = array[sorted_indices]
print(sorted_array)

Выход:

[[1 4 7]
 [2 5 8]
 [3 6 9]]

В этой статье мы рассмотрели три различных метода сортировки массивов по столбцам в Python. Мы обсудили использование встроенной функции sorted() с лямбда-функцией, использование функции itemgetter() из модуля оператора и использование функции argsort() из numpy для расширенных числовых операций. Применяя эти методы, вы можете эффективно сортировать массивы по определенным столбцам, удовлетворяя различные потребности программирования.