Question

我在一个名为ratings的pandas数据框中有一些看起来像这样的数据：

           rating                                                        
id       10063204 10073140 10077387 10091371 10124740 10136418 10146903
user_id
10003869      NaN      8.0      NaN      NaN      3.0      NaN      NaN
10022889      NaN      NaN      1.0      NaN      NaN      NaN      NaN

用于行的user_id，用于列的id和用作值的评级。我在互联网上看了一遍，找不到答案：

我使用的列是数字，8位数，目前它们是整数（或浮点数，无论哪种方式它们都不是字符串）但我不能搜索或选择给定id，除非它们是字符串。

使用loc / iloc不起作用，因为它们是列号。

例如，假设我有id：10146903（这里的最后一列，但真正的数据集有数千列，所以我不知道它的列号只是给了这个id）但我有数千列，我想要拉出该特定ID的列。

我是否必须将它们更改为字符串才能像那样搜索它们？或者有没有办法找到它，如果他们不是字符串？我宁愿不把它们转换成字符串，因为它们只需要稍后再次成为一个整体。

我以为我可以将这些数字用作列的索引，但似乎你只能在行上执行此操作。

我试图这样做：

specificID = ratings[10146903]

但那只是返回

KeyError: 'the label [10146903] is not in the [index]'

Answer 1

使用双方括号表示索引数组，例如df[[j,j]]然后i和j可以是整数。

示例：

>>> np.random.seed(0)
>>> df = pd.DataFrame(np.random.randn(8, 4),  columns=[1234, 56, 7, 890])
>>> df
       1234      56        7         890 
0  1.764052  0.400157  0.978738  2.240893
1  1.867558 -0.977278  0.950088 -0.151357
2 -0.103219  0.410599  0.144044  1.454274
3  0.761038  0.121675  0.443863  0.333674
4  1.494079 -0.205158  0.313068 -0.854096
5 -2.552990  0.653619  0.864436 -0.742165
6  2.269755 -1.454366  0.045759 -0.187184
7  1.532779  1.469359  0.154947  0.378163
>>> df[[56]]
         56
0  0.400157
1 -0.977278
2  0.410599
3  0.121675
4 -0.205158
5  0.653619
6 -1.454366
7  1.469359

pandas df可以为可以选择的列提供数字吗？

1 个答案: