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.
Python3
# 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
Python3
# 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
Python3
# 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
Python3
# 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:
Publicación traducida automáticamente
Artículo escrito por Chaitanya Tyagi y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA