主页 / 问题 / 1157976
Accepted
Vlad Vladov
Asked:2020-07-27 23:48:37 +0000 UTC2020-07-27 23:48:37 +0000 UTC

如何仅按前两位数字按列选择？

  • 772

有一个 DataFrame，其中Инн是一个索引（int）。
该表如下所示：

Инн        Название       Тип
231216     Орк            есть
432569     Шрек           нет

告诉我如何按Инн前两位数字按列进行选择（其余数字无关紧要）？

python

2 个回答

  1. 例子：

    search_items = [42, 43, 44]

res = df[df["Инн"].astype(str).str[:2].astype("int8").isin(search_items)]

    结果：

    In [22]: res
Out[22]:
      Инн Название  Тип
1  432569     Шрек  нет
    • 3
  2. Best Answer

    不将列转换Инн为行的解决方案：

    In [34]: mask = (df["Инн"] // 10**(np.ceil(np.log10(df["Инн"])).astype(int)-2)).isin(search_items)

In [35]: df[mask]
Out[35]:
      Инн Название  Тип
1  432569     Шрек  нет

    一步步：

    找出数字的位数：

    In [37]: np.ceil(np.log10(df["Инн"])).astype(int)
Out[37]:
0    6
1    6
Name: Инн, dtype: int32

    提高 10 次方(число разрядов минус 2)：

    In [38]: 10**(np.ceil(np.log10(df["Инн"])).astype(int)-2)
Out[38]:
0    10000
1    10000
Name: Инн, dtype: int32

    将原始数字系列除以（整数）上一步中获得的结果：

    In [39]: (df["Инн"] // 10**(np.ceil(np.log10(df["Инн"])).astype(int)-2))
Out[39]:
0    23
1    43
Name: Инн, dtype: int64

    ...

    • 2

相关问题