时间戳

首页 > 解决方案 > 如何在数据框中保留唯一的前 N ​​个值

python - 如何在数据框中保留唯一的前 N ​​个值

问题描述

它是pandas/Dataframe，对于每一行，我只想保留前N（N = 3）个值并将其他值设置为nan，

import pandas as pd
import numpy as np

data = np.array([['','day1','day2','day3','day4','day5'],
                ['larry',1,4,4,3,5],
                ['gunnar',2,-1,3,4,4],
                ['tin',-2,5,5, 6,7]])
                
df = pd.DataFrame(data=data[1:,1:],
                  index=data[1:,0],
                  columns=data[0,1:])
print(df)

输出是

       day1 day2 day3 day4 day5
larry     1    4    4    3    5
gunnar    2   -1    3    4    4
tin      -2    5    5    6    7

我想得到

       day1 day2 day3 day4 day5
larry   NaN    4    4  NaN    5
gunnar  NaN  NaN    3    4    4
tin     NaN    5  NaN    6    7

类似于pandas：只保留前 n 个值并将其他值设置为 0，但我只需要保留 N 个最高可用值，否则平均值不正确

对于上面的结果，我只想保留5第一

标签： pythonpandasdataframenumpy

解决方案

您可以使用np.unique排序并找到第 5 个最大值，然后使用where：

uniques = np.unique(df)

# what happens if len(uniques) < 5?
thresh = uniques[-5]
df.where(df >= thresh)

输出：

        day1  day2  day3  day4  day5
larry    NaN   4.0     4     3     5
gunnar   NaN   NaN     3     4     4
tin      NaN   5.0     5     6     7

更新：第二次看，我认为你可以这样做：

df.apply(pd.Series.nlargest, n=3,axis=1).reindex(df.columns, axis=1)

输出：

        day1  day2  day3  day4  day5
larry    NaN   4.0   4.0   NaN   5.0
gunnar   NaN   NaN   3.0   4.0   4.0
tin      NaN   5.0   NaN   6.0   7.0

推荐阅读