Question

我实现了一个函数，该函数会在熊猫数据框中首次出现某个值，但我觉得这种实现有点难看。您有更好的方法来实现它吗？

[mots]是一个字符串数组

# Sans doutes la pire implémentation au monde...
def find_singular_value(self, mots):
    bool_table = self.document.isin(mots)
    for i in range(bool_table.shape[0]):
        for j in range(bool_table.shape[1]):
            boolean = bool_table.iloc[i][j]
            if boolean:
                return self.document.iloc[i][j + 1]

Answer 1

这支班轮应该给您您所需要的。

self.document[self.document.isin(mots)].melt()["value"].dropna().values[0]

它将isin掩码应用于原始df，然后使用pd.melt和df.dropna找到第一个非nan值

这是一个简单的细分：

>>> df = pd.DataFrame({"a":[1,2,3],"b":[4,5,6],"c":[7,8,9]})
>>> df.isin([4,6])
       a      b      c
0  False   True  False
1  False  False  False
2  False   True  False
>>> df[df.isin([4,6])]
    a    b   c
0 NaN  4.0 NaN
1 NaN  NaN NaN
2 NaN  6.0 NaN
>>> df[df.isin([4,6])].melt()
  variable  value
0        a    NaN
1        a    NaN
2        a    NaN
3        b    4.0
4        b    NaN
5        b    6.0
6        c    NaN
7        c    NaN
8        c    NaN
>>> df[df.isin([4,6])].melt()["value"]
0    NaN
1    NaN
2    NaN
3    4.0
4    NaN
5    6.0
6    NaN
7    NaN
8    NaN
Name: value, dtype: float64
>>> df[df.isin([4,6])].melt()["value"].dropna()
3    4.0
5    6.0
Name: value, dtype: float64
>>> df[df.isin([4,6])].melt()["value"].dropna().values
array([ 4.,  6.])
>>> df[df.isin([4,6])].melt()["value"].dropna().values[0]
4.0
>>>

Answer 2

这是获取j + 1值的解决方案。它使用df.unstack和df.shift

df = self.document.unstack()
vals = df[df.isin(mots).shift().fillna(False)]

val将包含self.documents中的所有j + 1值。然后，您可以像我以前的答案一样选择第一个。希望这对您有用。

熊猫python：从DataFrame获取一个值

2 个答案: