In Apache Beam Python SDK, it is possible to perform the following:
input | GroupBy(account=lambda s: s["account&q