主页 / 问题 / 967312
Accepted
Pavel
Asked:2020-04-09 15:01:04 +0000 UTC2020-04-09 15:01:04 +0000 UTC

pandas列中值的分布

  • 772

有一个数据框：

id      date     score
12    15.05.19     4
34    10.04.19     3
17    02.06.19     7
19    30.07.19     5
25    28.05.19     6

有一份负责人名单：

spisok = ['Ivanov', 'Petrov', 'Sidorov']

我想得到结果 - 在数据框中添加一个包含负责人的列：

id      date     score   in_charge
12    15.05.19     4      Ivanov
34    10.04.19     3      Petrov
17    02.06.19     7      Sidorov
19    30.07.19     5      Ivanov
25    28.05.19     6      Petrov

我写了一个函数，但我不能应用它：

def an_distr(df, spisok):


    count = len(df)
    while count >= 0:
        for el in spisok:
            return el
            count = count - 1

df['in_charge'] = an_distr(df, spisok)

给出错误信息

TypeError：“函数”对象不可迭代

python

1 个回答

  1. Best Answer

    您的功能无法正常工作，如果只是因为之后

    return el

    该函数将返回一个值，并且不会执行以下代码。

    想到了以下解决方案，称为“额头上”：

    # Формируем список, который будет добавлен в качестве столбца
s = [spisok[i % len(spisok)] for i in range(len(df))]

df['in_charge'] = s

>>> df
id      date  score in_charge
12  15.05.19      4    Ivanov
34  10.04.19      3    Petrov
17  02.06.19      7   Sidorov
19  30.07.19      5    Ivanov
25  28.05.19      6    Petrov
    • 1

相关问题