如何有条件地从熊猫系列中选择物品

Question

我正在使用由数字列表组成的Pandas系列，单词作为索引：

$10             [1, 0, 1, 1, 1, 1, 1]
$100                        [0, 0, 0]
$15                               [1]
$19                            [0, 0]
$1?                            [1, 1]
$20                         [0, 1, 1]
$20-$40                           [0]

我正在尝试编写一些简单的代码，以创建一个新系列，其中仅包括包含长度为'n'或更大的列表的项目。

有点像系列的列表理解。

谢谢你的帮助

Answer 1

您应该避免在Series对象中使用list ，但是您可以按照以下要求进行操作：

编辑：用法

# DON'T use `eval` in production I'm just using it for convenience here
In [7]: s = read_clipboard(sep=r'\s{2,}', index_col=0, header=None, squeeze=1).map(eval)

In [8]: s
Out[8]:
0
$10        [1, 0, 1, 1, 1, 1, 1]
$100                   [0, 0, 0]
$15                          [1]
$19                       [0, 0]
$1?                       [1, 1]
$20                    [0, 1, 1]
$20-$40                      [0]

In [20]: n = 3

In [21]: s.map(len) >= n
Out[21]:
0
$10         True
$100        True
$15        False
$19        False
$1?        False
$20         True
$20-$40    False
Name: 1, dtype: bool

In [22]: s[s.map(len) >= n]
Out[22]:
0
$10     [1, 0, 1, 1, 1, 1, 1]
$100                [0, 0, 0]
$20                 [0, 1, 1]
Name: 1, dtype: object

您不应该在Series对象中使用list ，因为list是object数组，而不是可以利用numpy速度的同类类型Series 。

Answer 2

尝试这个：

s[s.map(len) >= n]