Início / coding / Perguntas / 77391325
Accepted
jo_
Asked: 2023-10-31 03:22:13 +0800 CST

Interpretando caso especial de loop 'x for x in ...' em python

  • 772

Tenho uma dúvida sobre o trecho de código abaixo:

for i, name in enumerate([x[:-3] for x in dm_cov.columns[:10]]):
    stats['beta'][i].write(f'../data/{sub}_beta_{name}.nii.gz')

É correto dizer que o trecho de código acima é funcionalmente equivalente ao trecho de código abaixo?

for i, name in enumerate([x for x in dm_cov.columns[:7]]):
    stats['beta'][i].write(f'../data/{sub}_beta_{name}.nii.gz')

Estou curioso porque xdeveria ser um nome de coluna, então não tenho certeza de como ele pode ser subdividido em [:-3].

Obrigado!

python

3 respostas

  1. Defina columnscomo iterável e compare as diferenças:

    columns = [
    'january', 'february', 'march', 'april',
    'may', 'june', 'july', 'august',
    'september', 'october', 'november', 'december',
]

for i, name in enumerate([x[:-3] for x in columns[:10]]):
    print(name)
    # janu
    # febru
    # ma
    # ap
    #
    # j
    # j
    # aug
    # septem
    # octo
    # > Only the first 10 elements, without the last 3 letters each
    
for i, name in enumerate([x for x in columns[:7]]):
    print(name)
    # january
    # february
    # march
    # april
    # may
    # june
    # july
    # > Only the first 7 elements
    • 3
  2. Best Answer

    Não, os dois trechos não são equivalentes. dm_cov.columns[:10]fornece os primeiros dez nomes de colunas.

    x[:-3] for x in ...em seguida, corta os três últimos caracteres de cada nome de coluna fornecido por dm_cov.columns[:10].

    Seu segundo snippet [x for x in dm_cov.columns[:7]]leva apenas os primeiros sete nomes de colunas na íntegra.

    A propósito, essa compreensão da lista é inútil no segundo trecho, você poderia simplesmenteenumerate(dm_cov.columns[:7])

    Você pode verificar isso usando um iterável diferente em vez de dm_cov.columns:

    test = [f"data{i}_col" for i in range(15)]
# ['data0_col', 'data1_col', 'data2_col', 'data3_col', 'data4_col', 'data5_col', 'data6_col', 'data7_col', 'data8_col', 'data9_col', 'data10_col', 'data11_col', 'data12_col', 'data13_col', 'data14_col']

print([x[:-3] for x in test[:10]])
# ['data0_', 'data1_', 'data2_', 'data3_', 'data4_', 'data5_', 'data6_', 'data7_', 'data8_', 'data9_']

print([x for x in test[:7]])
# ['data0_col', 'data1_col', 'data2_col', 'data3_col', 'data4_col', 'data5_col', 'data6_col']
    • 1

  3. Não, eles não são equivalentes.

    Se xfor um nome de coluna, x[:-3]é o nome da coluna com os últimos 3 caracteres removidos. A segunda versão do código não remove os últimos 3 caracteres.

    A primeira versão itera pelos primeiros 10 nomes de colunas, a segunda versão itera apenas pelos primeiros 7 nomes de colunas.

    Você parece ter assumido que as duas fatias estão operando no mesmo nível, então você pode combinar os índices finais. Mas uma fatia opera na lista de nomes, a outra opera em cada nome.

    • 1

relate perguntas