Методы объединения столбцов в единый список в Python

Чтобы превратить несколько столбцов в один столбец в виде списка в Python, вы можете использовать различные методы в зависимости от структуры данных, с которой вы работаете. Вот несколько возможных подходов:

  1. Использование pandas:
    Если вы работаете с DataFrame pandas, вы можете объединить несколько столбцов в один столбец с помощью функции pd.concat(). Вот пример:

    import pandas as pd
    # Create a DataFrame
    df = pd.DataFrame({'Column1': [1, 2, 3], 'Column2': [4, 5, 6], 'Column3': [7, 8, 9]})
    # Concatenate columns into a single column as a list
    df['Combined'] = df.apply(lambda row: [row['Column1'], row['Column2'], row['Column3']], axis=1)
    # Print the resulting DataFrame
    print(df)
  2. Использование zip и понимания списков.
    Если у вас есть несколько списков или столбцов, вы можете использовать функцию zip()в сочетании с пониманием списков, чтобы объединить их в один список.. Вот пример:

    # Create example lists
    list1 = [1, 2, 3]
    list2 = [4, 5, 6]
    list3 = [7, 8, 9]
    # Combine lists into a single list using zip and list comprehension
    combined_list = [list(row) for row in zip(list1, list2, list3)]
    # Print the combined list
    print(combined_list)
  3. Использование numpy:
    Если вы работаете с массивами numpy, вы можете использовать функцию numpy.concatenate()для объединения нескольких столбцов в один. Вот пример:

    import numpy as np
    # Create example numpy arrays
    array1 = np.array([1, 2, 3])
    array2 = np.array([4, 5, 6])
    array3 = np.array([7, 8, 9])
    # Concatenate arrays into a single array using numpy.concatenate()
    combined_array = np.concatenate((array1, array2, array3))
    # Convert the combined array to a list
    combined_list = combined_array.tolist()
    # Print the combined list
    print(combined_list)