Created
March 11, 2020 22:02
-
-
Save luminoso/385d0193ac7e2576be349bed63a88056 to your computer and use it in GitHub Desktop.
koalas itertuples wrapper for iterrows
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# koalas don't support iterrows. so we convert iterrows() to itertuples() format | |
if internal_itertuples: | |
Row = namedtuple('Row', df.columns) | |
def convert_iterrow_to_itertuples(df): | |
for row in df.iterrows(): | |
yield Row(*row[1]) | |
itertuples = convert_iterrow_to_itertuples(df) | |
else: | |
itertuples = df.itertuples() | |
for i in intertuples: | |
print(i) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment