Início / coding / Perguntas / 78275265
Accepted
AmirX
Asked: 2024-04-05 00:26:00 +0800 CST

Existe uma maneira ONE-LINER de fornecer a cada linha de um dataframe um ID exclusivo que consiste em um número inteiro e uma string?

  • 772

Este é o meu DataFrame:

import pandas as pd
df = pd.DataFrame(
    {
        'a': [4, 3, 2, 2, 6]
    }
)

E esta é a saída esperada: quero criar uma coluna id:

   a   id
0  4  x_0
1  3  x_1
2  2  x_2
3  2  x_3
4  6  x_4

Posso criar idassim, mas acho que existe uma linha única para isso:

df['id'] = np.arange(len(df))
df['id'] = 'x_' + df.id.astype(str)

Eu prefiro uma solução que não use index.

python

2 respostas

  1. Provavelmente faz sentido usar indexaqui:

    df['id'] = "x_" + df.index.to_frame(name='id').astype(str)

print(df)

   a   id
0  4  x_0
1  3  x_1
2  2  x_2
3  2  x_3
4  6  x_4
    • 2
  2. Best Answer

    Se você quiser um one-liner, terá que converter seu array em Series (arrays brutos não são práticos para concatenação de strings):

    df['id'] = 'x_' + pd.Series(np.arange(len(df)), dtype=str)

# or
df['id'] = pd.Series(np.arange(len(df)), dtype=str).radd('x_')

    Saída:

       a   id
0  4  x_0
1  3  x_1
2  2  x_2
3  2  x_3
4  6  x_4
    • 1

relate perguntas