主页 / coding / 问题 / 79054792
Accepted
Giampaolo Levorato
Asked: 2024-10-04 22:47:44 +0800 CST

应用以 gt/lt 条件为键的 pandas 词典

  • 772

我创建了以下熊猫数据框：

ds = {'col1':[1,2,2,3,4,5,5,6,7,8]}

df = pd.DataFrame(data=ds)

数据框如下所示：

print(df)

   col1
0     1
1     2
2     2
3     3
4     4
5     5
6     5
7     6
8     7
9     8

然后我创建了一个名为的新字段newCol，其定义如下：

def criteria(row):
    if((row['col1'] > 0) & (row['col1'] <= 2)):
        return "A"
    elif((row['col1'] > 2) & (row['col1'] <= 3)):
        return "B"
    else:
        return "C"
    
df['newCol'] = df.apply(criteria, axis=1)

新的数据框如下所示：

print(df)

   col1 newCol
0     1      A
1     2      A
2     2      A
3     3      B
4     4      C
5     5      C
6     5      C
7     6      C
8     7      C
9     8      C

是否有可能创建这样的字典：

dict = {
        
        '0 <= 2' : "A",
        '2 <= 3' : "B",
        'Else' : "C"

        }

然后将其应用到数据框：

df['newCol'] = df['col1'].map(dict)

？

有人能帮帮我吗？

python

1 个回答

  1. Best Answer

    是的，你可以这样做IntervalIndex：

    dic = {(0, 2): 'A',
       (2, 3): 'B',
      }
other = 'C'

bins = pd.IntervalIndex.from_tuples(dic)
labels = list(dic.values())

df['newCol'] = (pd.Series(labels, index=bins)
                  .reindex(df['col1']).fillna(other)
                  .tolist()
               )

    但考虑到你的例子，似乎更直接的做法是cut：

    df['newCol'] = pd.cut(df['col1'], bins=[0, 2, 3, np.inf], labels=['A', 'B', 'C'])

    输出：

       col1 newCol
0     1      A
1     2      A
2     2      A
3     3      B
4     4      C
5     5      C
6     5      C
7     6      C
8     7      C
9     8      C

    如果你坚持使用原来的字典格式，你可以使用以下方法进行转换：

    dic = {'0 <= 2' : "A",
       '2 <= 3' : "B",
       'Else' : "C"
}

dic2 = {tuple(map(int, k.split(' <= '))): v for k, v in dic.items()
        if k != 'Else'}
# {(0, 2): 'A', (2, 3): 'B'}
other = dic['Else']
    • 1

相关问题