Agregar nueva columna a DataFrame existente en Pandas

Analicemos cómo agregar nuevas columnas al DataFrame existente en Pandas. Hay varias maneras en que podemos hacer esta tarea.

Método #1: declarando una nueva lista como una columna. 

# Import pandas package
import pandas as pd
  
# Define a dictionary containing Students data
data = {'Name': ['Jai', 'Princi', 'Gaurav', 'Anuj'],
        'Height': [5.1, 6.2, 5.1, 5.2],
        'Qualification': ['Msc', 'MA', 'Msc', 'Msc']}
  
# Convert the dictionary into DataFrame
df = pd.DataFrame(data)
  
# Declare a list that is to be converted into a column
address = ['Delhi', 'Bangalore', 'Chennai', 'Patna']
  
# Using 'Address' as the column name
# and equating it to the list
df['Address'] = address
  
# Observe the result
print(df)

Producción: 

Tenga en cuenta que la longitud de su lista debe coincidir con la longitud de la columna de índice; de ​​lo contrario, mostrará un error. 

Método #2: Usando DataFrame.insert()

Da la libertad de agregar una columna en cualquier posición que deseemos y no solo al final. También proporciona diferentes opciones para insertar los valores de la columna.

Ejemplo

# Import pandas package 
import pandas as pd
  
# Define a dictionary containing Students data
data = {'Name': ['Jai', 'Princi', 'Gaurav', 'Anuj'],
        'Height': [5.1, 6.2, 5.1, 5.2],
        'Qualification': ['Msc', 'MA', 'Msc', 'Msc']}
  
# Convert the dictionary into DataFrame
df = pd.DataFrame(data)
  
# Using DataFrame.insert() to add a column
df.insert(2, "Age", [21, 23, 24, 21], True)
  
# Observe the result
print(df)

Producción: 

Método #3: Usar el método Dataframe.assign()

Este método creará un nuevo marco de datos con una nueva columna agregada al marco de datos anterior.

Ejemplo

# Import pandas package
import pandas as pd
  
# Define a dictionary containing Students data
data = {'Name': ['Jai', 'Princi', 'Gaurav', 'Anuj'],
        'Height': [5.1, 6.2, 5.1, 5.2],
        'Qualification': ['Msc', 'MA', 'Msc', 'Msc']}
  
  
# Convert the dictionary into DataFrame
df = pd.DataFrame(data)
  
# Using 'Address' as the column name and equating it to the list
df2 = df.assign(address=['Delhi', 'Bangalore', 'Chennai', 'Patna'])
  
# Observe the result
print(df2)

Producción: 

Método #4: Usando un diccionario

Podemos usar un diccionario de Python para agregar una nueva columna en pandas DataFrame. Utilice una columna existente como valores clave y sus respectivos valores serán los valores para una nueva columna.

Ejemplo

# Import pandas package
import pandas as pd
  
# Define a dictionary containing Students data
data = {'Name': ['Jai', 'Princi', 'Gaurav', 'Anuj'],
        'Height': [5.1, 6.2, 5.1, 5.2],
        'Qualification': ['Msc', 'MA', 'Msc', 'Msc']}
  
# Define a dictionary with key values of
# an existing column and their respective
# value pairs as the # values for our new column.
address = {'Delhi': 'Jai', 'Bangalore': 'Princi',
           'Patna': 'Gaurav', 'Chennai': 'Anuj'}
  
# Convert the dictionary into DataFrame
df = pd.DataFrame(data)
  
# Provide 'Address' as the column name
df['Address'] = address
  
# Observe the output
print(df)

Producción: