如何添加基于熊猫中另一列的值的新列

时间:2020-05-30 03:07:06

标签: python pandas dataframe

让我们说我有以下df:

rs = np.random.RandomState(1979)
x = rs.randn(500)
g = np.tile(list("ABCDEFGHIJ"), 50)
df = pd.DataFrame(dict(x=x, g=g))
m = df.g.map(ord)
df["x"] += m

df

           x    g
0   64.038123   A
1   66.147050   B
2   66.370011   C
3   68.791019   D
4   70.583534   E
... ... ...
495 69.358022   F
496 72.212877   G
497 70.474247   H
498 73.251022   I
499 74.461828   J

我想添加一个新列,该列由与字母相匹配的数字1-10组成,如下所示:

g new
A  1
B  2
C  3
D  4

我正在使用的实际数据使用名称而不是字母,但是我显然不想在这里使用它们。但是,我的目标还是要创建一个新列,其中包含我想要的某些名称的值。

谢谢!

1 个答案:

答案 0 :(得分:3)

听起来像是字典,将名称映射到您想要在图形中使用的相应顺序位置是可行的方式。这是一个简单的示例:

>>> import pandas as pd
>>> df = pd.DataFrame({"Name": ["Alice", "Bob", "Bob", "Alice", "Charlie"]})
>>> ordering = {"Alice": 1, "Bob": 3, "Charlie": 2}
>>> df["Ordering"] = df["Name"].map(ordering)
>>> df
      Name  Ordering
0    Alice         1
1      Bob         3
2      Bob         3
3    Alice         1
4  Charlie         2