Eliminar una columna CSV en Python

El archivo de valores separados por comas (CSV) es un archivo de texto delimitado que utiliza comas para valores individuales. Cada línea del archivo es un registro de datos en CSV. Este formato se usa para datos tabulares, filas y columnas, exactamente como una hoja de cálculo. El archivo CSV almacena datos en filas y los valores en cada fila se separan con una coma (separador), también conocido como delimitador.

Hay 2 formas de eliminar una columna por completo de un CSV en python. Centrémonos ahora en las técnicas:

1. Con la biblioteca de pandas: drop() o pop()
2. Sin biblioteca de pandas

Aquí, se utiliza un archivo CSV simple, es decir; entrada.csv

Método 1: usar la biblioteca de pandas

Pandas Pandas consta de una función de eliminación que se utiliza para eliminar filas o columnas de los archivos CSV. Pop() el pop()

Ejemplo 1: Usar drop()

data.drop( labels=None, axis=0, index=None, columns=None, level=None, inplace=False,errors='raise')

1. Importar pandas
2. Leer archivo CSV
3. Use la función drop() para eliminar o eliminar filas o columnas de los archivos CSV
4. Imprimir datos

# import pandas with shortcut 'pd'
import pandas as pd  
  
# read_csv function which is used to read the required CSV file
data = pd.read_csv('input.csv')
  
# display 
print("Original 'input.csv' CSV Data: \n")
print(data)
  
# drop function which is used in removing or deleting rows or columns from the CSV files
data.drop('year', inplace=True, axis=1)
  
# display 
print("\nCSV Data after deleting the column 'year':\n")
print(data)

Producción:

Ejemplo 2: Usando pop()

Podemos usar el método panda pop() para eliminar columnas de CSV nombrando la columna como argumento.

1. Importar pandas
2. Leer archivo CSV
3. Use la función pop() para eliminar o eliminar filas o columnas de los archivos CSV
4. Imprimir datos

# import pandas with shortcut 'pd'
import pandas as pd
  
# read_csv function which is used to read the required CSV file
data = pd.read_csv('input.csv')
  
# display
print("Original 'input.csv' CSV Data: \n")
print(data)
  
# pop function which is used in removing or deleting columns from the CSV files
data.pop('year')
  
# display
print("\nCSV Data after deleting the column 'year':\n")
print(data)

Producción:

Método 2: usar la biblioteca CSV

Ejemplo 3: Uso de lectura y escritura de CSV

1. Abrir archivo CSV de entrada como fuente
2. Leer archivo CSV de origen
3. Abrir archivo CSV de salida como resultado
4. Ponga los datos CSV de origen en el CSV de resultados usando índices

# import csv
import csv
  
# open input CSV file as source
# open output CSV file as result
with open("input.csv", "r") as source:
    reader = csv.reader(source)
      
    with open("output.csv", "w") as result:
        writer = csv.writer(result)
        for r in reader:
            
            # Use CSV Index to remove a column from CSV
            #r[3] = r['year']
            writer.writerow((r[0], r[1], r[2], r[4], r[5]))

Producción: