主页 / coding / 问题 / 77166543
Accepted
Pro Q
Asked: 2023-09-24 17:57:54 +0800 CST

df 列中的数字，但不在该列的列表版本中

  • 772

我有以下代码：

    if 0 in df[RATING_COL]:
        rating_col_list = df[RATING_COL].to_list()
        assert 0 in rating_col_list

断言正在触发一个AssertionError. 这怎么可能？为什么列中有一个 0，但是当我将列转换为列表时，0 消失了？

我正在加载的数据帧基于 MovieLens-1M，如下所示：

user_id,item_id,rating
1,1193000,2
1,1193001,3
1,1193002,4
1,1193003,5
1,1193004,6
1,1193005,7
1,1193006,8
1,1193007,9
1,1193008,10
1,661000,6
1,661001,7
1,661002,8
1,661003,9
1,661004,10
1,661005,9
1,661006,8
1,661007,7
1,661008,6

在此格式中，1,1193008,10表示用户 1 对项目 1193 的评分为 8。10 表示这是评分，以 1193 开头的所有其他项目的评分将低于 10。（因此表示用户 1 对项目 661 的评分为 8）1,661004,10。 4.)

（另外，我已经用 CTRL-F 检查过：评级栏中没有 0 评级。）

python

2 个回答

  1. 栏里怎么会有0呢..？

    因为0是Series的一个索引df[RATING_COL]。

    RATING_COL = "rating"

print(df[RATING_COL])

0      2 # <-- index 0
1      3
2      4
3      5
4      6
5      7
6      8
7      9
8     10
9      6
10     7
11     8
12     9
13    10
14     9
15     8
16     7
17     6
Name: rating, dtype: int64

    这就像检查某个键是否在字典中时，O in d将检查是否0是 的键之一d。

    • 3
  2. Best Answer

    0 in df[RATING_COL]在索引中搜索，要在系列值中搜索，请使用：

    if 0 in df[RATING_COL].values:
    rating_col_list = df[RATING_COL].to_list()
    assert 0 in rating_col_list
    • 3

相关问题