擅长:python、mysql、java
<p>您可以通过<code>get_duplicates()</code>创建一个<code>list</code>副本,然后通过<code>A</code>的索引设置<code>dgroup_id</code></p>
<pre><code>def find_index(string):
if string in duplicates:
return duplicates.index(string)+1
else:
return 0
df = pd.DataFrame({'A': ('foo', 'bar', 'foo', 'baz', 'foo', 'bar')})
duplicates = df.set_index('A').index.get_duplicates()
df['dgroup_id'] = df['A'].apply(find_index)
df
</code></pre>
<p>输出:</p>
^{pr2}$