¿Cómo obtener la primera columna de Pandas DataFrame?

En este artículo, discutiremos cómo obtener la primera columna del marco de datos de pandas en el lenguaje de programación Python.

Método 1: Usar la función iloc[]

Esta función se usa para obtener la primera columna usando el operador de división. para las filas las extraemos todas, para las columnas especificamos el índice de la primera columna.

Sintaxis :

dataframe.iloc[:, 0]

dónde. dataframe es el dataframe de entrada

La operación de corte será como:

[inicio_fila:fin_fila , inicio_columna, fin_columna]

dónde,

• row_start se refiere a la fila de inicio con la posición 0 como índice
• row_end se refiere a la última fila con la posición n como índice
• column_start se refiere a la columna de inicio con la posición 0 como índice
• column_end se refiere a la última columna con la posición n como índice

También se puede emplear la siguiente alternativa.

Sintaxis:

dataframe.iloc[:, :1]

dónde. dataframe es el dataframe de entrada. Ambos devolverán el tipo de datos del marco de datos.

Ejemplo: programa de Python para obtener la primera columna utilizando los enfoques anteriores

# import pandas module
import pandas as pd
  
# create dataframe with 3 columns
data = pd.DataFrame({
    "id": [7058, 7059, 7072, 7054],
    "name": ['sravan', 'jyothika', 'harsha', 'ramya'],
    "subjects": ['java', 'python', 'html/php', 'php/js']
}
)
  
# display dataframe
print(data)
  
print("--------------")
  
# get first column by returning series
print(data.iloc[:, 0])
  
print("--------------")
  
# get first column by returning dataframe
print(data.iloc[:, :1])

Producción:

Método 2: Usar columnas[]

Este método devolverá la columna basada en el índice. Entonces, tenemos que dar 0 para obtener la primera columna.

Sintaxis :

marco de datos[marco de datos.columnas[0]]

dónde

• dataframe es el dataframe de entrada
• las columnas [0] representan la primera columna

Ejemplo : programa de Python para obtener la primera columna usando columnas []

# import pandas module
import pandas as pd
  
# create dataframe with 3 columns
data = pd.DataFrame({
    "id": [7058, 7059, 7072, 7054],
    "name": ['sravan', 'jyothika', 'harsha', 'ramya'],
    "subjects": ['java', 'python', 'html/php', 'php/js']
}
)
  
# display dataframe
print(data)
  
print("--------------")
  
  
# get first column by returning dataframe
# using columns[] method
print(data[data.columns[0]])

Producción:

Método 3: usar el nombre de la columna

Podemos usar el nombre de la primera columna para obtener la primera columna.

Sintaxis :

dataframe.first_column

Ejemplo: código de Python para obtener la primera columna usando el nombre de la columna

# import pandas module
import pandas as pd
  
# create dataframe with 3 columns
data = pd.DataFrame({
    "id": [7058, 7059, 7072, 7054],
    "name": ['sravan', 'jyothika', 'harsha', 'ramya'],
    "subjects": ['java', 'python', 'html/php', 'php/js']
}
)
  
# display dataframe
print(data)
  
print("--------------")
  
  
# get first column by returning dataframe
# using column_name
print(data.id)

Producción:

También podemos usar la función head() dentro de esto para mostrar el número de filas en la primera columna.

Ejemplo: código de Python para obtener la primera columna usando el nombre de la columna

# import pandas module
import pandas as pd
  
# create dataframe with 3 columns
data = pd.DataFrame({
    "id": [7058, 7059, 7072, 7054],
    "name": ['sravan', 'jyothika', 'harsha', 'ramya'],
    "subjects": ['java', 'python', 'html/php', 'php/js']
}
)
  
# display dataframe
print(data)
  
print("--------------")
  
  
# get first column by returning dataframe
# using column_name
# display 1 row
print(data.id.head(1))
print("--------------")
  
# get first column by returning dataframe
# using column_name
# display 2 rows
print(data.id.head(2))
print("--------------")
  
# get first column by returning dataframe
# using column_name
# display 4 rows
print(data.id.head(4))

Producción:

Método 4: Usar la función head()

Esta función por defecto devuelve las filas superiores del marco de datos. Para devolver la columna, tenemos que transponer (intercambiar filas por columnas) el marco de datos usando la función T y obtener la primera columna.

Sintaxis :

dataframe.T.head(1).T

Ejemplo: código de Python para obtener la primera columna usando el nombre de la columna

# import pandas module
import pandas as pd
  
# create dataframe with 3 columns
data = pd.DataFrame({
    "id": [7058, 7059, 7072, 7054],
    "name": ['sravan', 'jyothika', 'harsha', 'ramya'],
    "subjects": ['java', 'python', 'html/php', 'php/js']
}
)
  
# display dataframe
print(data)
  
print("--------------")
# using head
print(data.T.head(1).T)

Producción: