PySpark无法访问使用StringIndexer添加的列

PySpark无法访问使用StringIndexer添加的列
我试图将String列转换Country为Interger列Country_ID，结果看起来很好。但当我试图访问Country_ID我得到的列AnalysisException。

以下是数据帧：

这是代码：

from pyspark.ml.feature import StringIndexer
indexer = StringIndexer(inputCol='Country', outputCol='Country_ID')
modified_df = indexer.fit(df).transform(df)
修改后的DataFrame：

modified_df.select('*').show()

过滤查询：

modified_df.select('UserId').filter(df['Country_ID'] == 2).show()
以下是例外情况：

AnalysisException: u'Cannot resolve column name "Country_ID" among (UserId, Country);'

我可以将列视为DataFrame的一部分

modified_df.columns
给，

['UserId', 'Country', 'Country_ID']
如何使DataFrame过滤条件有效Country_ID？