mars-project / mars

Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.
https://mars-project.readthedocs.io
Apache License 2.0
2.7k stars 326 forks source link

Support DataFrame indexing with mars Tensor #2440

Open MaxwellLZH opened 3 years ago

MaxwellLZH commented 3 years ago

Currently the following snippets causes error, which is quite surprising for users coming from Pandas background:

import mars.dataframe as md
import mars.tensor as mt
df = md.DataFrame(mt.random.randn(10, 4))
df.loc[df.index].execute()

It will be nice to support indexing with Mars Tensor as well.

Thank you !

qinxuye commented 3 years ago

Thanks, will support it ASAP.

MaxwellLZH commented 2 years ago

Wondering if the issue has been fixed in the newest version?