对于我来说,使用您的解决方案,替代方案是通过比较
True
:
allData = pd.DataFrame({'isForecast':[True, False, True]})
forecastOutput = allData.query('isForecast').copy()
print(forecastOutput)
isForecast
0 True
2 True
forecastOutput = allData.query('isForecast == True').copy()
print(forecastOutput)
isForecast
0 True
2 True
或布尔索引:
forecastOutput = allData[allData['isForecast']]
print(forecastOutput)
isForecast
0 True
2 True
如果值不是布尔值:
allData = pd.DataFrame({'isForecast':['True', 'False', 'True']})
forecastOutput = allData.query('isForecast == "True"').copy()
print(forecastOutput)
isForecast
0 True
2 True