由列表索引的python数据框

Question

我正在尝试获取包含来自有限集合的重复值的 DataFrame 列，并用索引号替换这些值，因此如果值为 [200,20,1000,1]，则它们出现的索引将为 [1 ,2,3,4]。 实际数据示例为：

0    aaa
1    aaa
2    bbb
3    aaa
4    bbb
5    bbb
6    ccc
7    ddd
8    ccc
9    ddd

想要的输出是

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

我想更改对数字毫无意义的值。就是这样......我不关心索引的顺序，即1可以是3等等，只要顺序是一致的。即，我不在乎 ['aaa','bbb','ccc','ddd'] 是否会被 [1,2,3,4] 或 [2,4,3,1] 索引。

假设 DF 名称是 tbl，我只想更改列“aaa”中的索引子集。让我们用 tbl_ind 来表示这些索引。我想要这样做的方式是：

tmp_r = tbl[tbl_ind]
un_r_ind = np.unique(tmp_r)
for r_ind in range(len(un_r_ind)):
    r_ind_ind = np.array(np.where(tmp_r == un_r_ind[r_ind])[0])
    for j_ind in range(len(r_ind_ind)):
        tbl['aaa'].iloc[tbl_ind[r_ind_ind[j_ind]]] = r_ind

它有效。而且在大型数据集上真的很慢。 Python 不允许更新tbl['aaa'].iloc[tbl_ind[r_ind_ind]]，因为它是一个索引列表...... 请帮忙？怎么可能加快这个速度？ 非常感谢！

Answer 1

您可以将rank 与dense 方法一起使用：

>>> df[0].rank("dense")
0    1
1    1
2    2
3    1
4    2
5    2
6    3
7    4
8    3
9    4
Name: 0, dtype: float64

这基本上是对值进行排序并将最低值映射到 1，将次低值映射到 2，依此类推。

Answer 2

我不确定我是否从您的示例中理解正确。 这是你想要达到的目标吗？ （除了索引上的偏差（零而不是一））：

df=['aaa','aaa','bbb','aaa','bbb','bbb','ccc','ddd','ccc','ddd']
idx={}

def index_data(v):
    global idx

    if v in idx:
        return idx[v]
    else:
        n = len(idx)
        idx[v] = n
        return n

if __name__ == "__main__":
    outlist = []
    for i in df:
        outlist.append(index_data(i))
    for i, v in enumerate(outlist):
        print i, v

它输出：

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

显然它可以被优化（例如，简单地为 n 增加一个计数器而不是检查索引的大小）

Answer 3

我会构造一个您要替换的值的字典，然后调用map：

In [7]:

df
Out[7]:
  data
0     
1  aaa
2  bbb
3  aaa
4  bbb
5  bbb
6  ccc
7  ddd
8  ccc
9  ddd
In [8]:

d = {'aaa':1,'bbb':2,'ccc':3,'ddd':4}
df['data'] = df['data'].map(d)
df

Out[8]:
   data
0      
1     1
2     2
3     1
4     2
5     2
6     3
7     4
8     3
9     4