如何使此代码更高效,即一行或两行?

时间:2018-04-28 17:27:06

标签: python pandas dataframe

$Matches.digita

我知道必须有一种方法可以将其压缩为使用的一行代码。我会使用“&”吗?符号连接前三行?

1 个答案:

答案 0 :(得分:1)

使用' query()'可以节省很多空间。

census_df = census_df.query("                       \
                (1 <= REGION <= 2)                  \
              &  CTYNAME.str.match('Washington')    \ 
              & (POPESTIMATE2015 > POPESTIMATE2014)
          ")[['STNAME', 'CTYNAME']]

在一行中:

census_df = census_df.query("(1 <= REGION <= 2) & CTYNAME.str.match('Washington') & (POPESTIMATE2015 > POPESTIMATE2014)")[['STNAME', 'CTYNAME']]