Question

我正在为软件包编写支票。 pandas DataFrame中的元素满足某些条件;如果他们不这样做，我将提出在Python中编写的ValueError异常。

以下是pandas DataFrame示例：

import pandas as pd
import numpy as np

dict1 = {'file': ['filename2', 'filename2', 'filename3', 'filename4', 
         'filename4', 'filename3'], 'amount': [3, 4, 5, 1, 2, 1], 
         'front': [21889611, 36357723, 196312, 11, 42, 1992], 
         'back':[21973805, 36403870, 277500, 19, 120, 3210], 
         'type':['A', 'A', 'A', 'B', 'B', 'C']}

df1 = pd.DataFrame(dict1)
print(df1)

        file  amount     front      back type
0  filename2       3  21889611  21973805    A
1  filename2       4  36357723  36403870    A
2  filename3       5    196312    277500    A
3  filename4       1        11        19    B
4  filename4       2        42       120    B
5  filename3       1      1992      3210    C

我看到检查某些值的最有效方法是使用sets，例如如果列type包含的元素不是A，B或C，则会抛出错误：

if not set(['A', 'B', 'C']).issubset(df1['type']):
    raise ValueError('Pandas DataFrame contains improper values in "type" column')

问题：

我如何最有效地检查条件？例如我想检查列amount是否包含大于0的整数。如果此列中有任何零，负整数或非整数，则引发ValueError()。

Answer 1

您可以只过滤列，获取返回的数据帧的长度，并在if语句中使用它：

len(df1[df1['amount'] > 0])

编写检查所有pandas DataFrame列值是否满足某个值？

1 个答案: