Como criar um dataframe 3d do pandas (com exemplo)

Você pode usar o módulo xarray para criar rapidamente um DataFrame 3D do pandas.

Este tutorial explica como criar o seguinte DataFrame 3D do pandas usando as funções do módulo xarray:

 product_A product_B product_C
year quarter                                 
2021 Q1 1.624345 0.319039 50
     Q2 -0.611756 0.319039 50
     Q3 -0.528172 0.319039 50
     Q4 -1.072969 0.319039 50
2022 Q1 0.865408 -0.249370 50
     Q2 -2.301539 -0.249370 50
     Q3 1.744812 -0.249370 50
     Q4 -0.761207 -0.249370 50

Exemplo: Crie um DataFrame 3D do Pandas

O código a seguir mostra como criar um conjunto de dados 3D usando funções xarray e NumPy :

 import numpy as np
import xarray as xr

#make this example reproducible
n.p. random . seeds (1)

#create 3D dataset
xarray_3d = xr. Dataset (
    { " product_A ": (("year", "quarter"), np.random.randn (2,4))},
    coordinates={
        " year ": [2021, 2022],
        " quarter ": ["Q1", "Q2", "Q3", "Q4"],
        " product_B ": ("year", np. random . randn (2)),
        " product_C ": 50,
    },
)

#view 3D dataset
print (xarray_3d)

Dimensions: (year: 2, quarter: 4)
Coordinates:
  * year (year) int32 2021 2022
  * quarter (quarter) <U2 'Q1' 'Q2' 'Q3' 'Q4'
    product_B (year) float64 0.319 -0.2494
    product_C int32 50
Data variables:
    product_A (year, quarter) float64 1.624 -0.6118 -0.5282 ... 1.745 -0.7612

Nota : A função NumPy randn() retorna valores de exemplo da distribuição normal padrão .

Podemos então usar a função to_dataframe() para converter este conjunto de dados em um DataFrame do pandas:

 #convert xarray to DataFrame
df_3d = xarray_3d. to_dataframe ()

#view 3D DataFrame
print (df_3d)

              product_A product_B product_C
year quarter                                 
2021 Q1 1.624345 0.319039 50
     Q2 -0.611756 0.319039 50
     Q3 -0.528172 0.319039 50
     Q4 -1.072969 0.319039 50
2022 Q1 0.865408 -0.249370 50
     Q2 -2.301539 -0.249370 50
     Q3 1.744812 -0.249370 50
     Q4 -0.761207 -0.249370 50

O resultado é um DataFrame 3D do pandas que contém informações sobre o número de vendas realizadas de três produtos diferentes em dois anos diferentes e quatro trimestres diferentes por ano.

Podemos usar a função type() para confirmar se este objeto é de fato um DataFrame do pandas:

 #display type of df_3d
type (df_3d)

pandas.core.frame.DataFrame

O objeto é de fato um DataFrame do pandas.

Recursos adicionais

Os tutoriais a seguir explicam como executar outras funções comuns em pandas:

Pandas: como encontrar valores únicos em uma coluna
Pandas: como encontrar a diferença entre duas linhas
Pandas: Como contar valores ausentes no DataFrame