Методы удаления столбцов в Python: примеры Pandas, NumPy и SQL

Под «Удалить столбцы» подразумевается процесс удаления определенных столбцов из набора данных. Вот несколько способов выполнить эту задачу, а также примеры кода на Python:

  1. Использование Pandas:

    import pandas as pd
    # Create a DataFrame
    df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
    # Drop a single column
    df.drop('B', axis=1, inplace=True)
    # or df = df.drop('B', axis=1)
    # Drop multiple columns
    df.drop(['B', 'C'], axis=1, inplace=True)
    # or df = df.drop(['B', 'C'], axis=1)
    print(df)

    Выход:

    A
    0  1
    1  2
    2  3
  2. Использование NumPy:

    import numpy as np
    # Create a 2D array
    arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
    # Drop a single column
    arr = np.delete(arr, 1, axis=1)
    # Drop multiple columns
    arr = np.delete(arr, [1, 2], axis=1)
    print(arr)

    Выход:

    [[1]
    [4]
    [7]]
  3. Использование SQL (при работе с базой данных):

    import sqlite3
    # Connect to the database
    conn = sqlite3.connect('your_database.db')
    cursor = conn.cursor()
    # Drop a single column
    cursor.execute("ALTER TABLE your_table DROP COLUMN column_name;")
    # Drop multiple columns
    cursor.execute("ALTER TABLE your_table DROP COLUMN column1, DROP COLUMN column2;")
    # Commit the changes
    conn.commit()