Skip to content

Instantly share code, notes, and snippets.

@luminoso
Created March 11, 2020 22:02
Show Gist options
  • Save luminoso/385d0193ac7e2576be349bed63a88056 to your computer and use it in GitHub Desktop.
Save luminoso/385d0193ac7e2576be349bed63a88056 to your computer and use it in GitHub Desktop.
koalas itertuples wrapper for iterrows
# 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