PySpark计数groupby和None键

Question

我有一个spark RDD对象(使用pyspark(，我正在尝试获得SQL的

SELECT MY_FIELD COUNT(*) GROUP BY MY_FIELD

所以我尝试了以下代码：

my_groupby_count = myRDD.map(lambda x: x.type).reduceByKey(lambda x, y: x + y).collect()
# 'type' is the name of the field inside the RDD row

但我收到了一个错误，我不知道如何处理：

---------------------------------------------------------------------------
Py4JJavaError                             Traceback (most recent call last)
<ipython-input-24-23b4c60c6fd6> in <module>()
----> 1 my_groupby_count = myRDD.map(lambda x: x.type).reduceByKey(lambda x, y: x +     y).collect()
/root/spark/python/pyspark/rdd.py in collect(self)

with SCCallSiteSync(self.context) as css:
-->              port = self.ctx._jvm.PythonRDD.collectAndServe(self._jrdd.rdd())
return list(_load_from_socket(port, self._jrdd_deserializer))

/root/spark/python/lib/py4j-0.10.4-src.zip/py4j/java_gateway.py in __call__(self, *args)
answer = self.gateway_client.send_command(command)
return_value = get_return_value(
->              answer, self.gateway_client, self.target_id, self.name)

现在，由于这种方法以前对我很有效，我怀疑它可能与数据本身有关。例如，我知道x.type中的一些值是None，但我不知道如何去掉它们。

有什么想法可以继续调查吗？P.S。toDF((也失败了，我想是由于同样的原因。此外，我会为RDD而不是DataFrame提供解决方案。感谢

Answer 1

您需要提供元组reduceByKey。看起来你刚刚忘记了"((">

myRDD.map(lambda x: (x.type, 1)).reduceByKey(lambda x, y: x + y).collect()

旁注：使用countByKey((的相同代码的较短版本

myRDD.map(lambda x: (x.type,)).countByKey()