mars-project / mars-project/mars

Support DataFrame indexing with mars Tensor

Open
#2,440 2 comments 0 reactions 0 assignees View on GitHub
mod: dataframe pr welcome prio: high task: medium type: feature
Dominant language
Python
Stars
2.7k
Forks
325
PR merge metrics
No merged PRs in 30d

Description

Currently the following snippets causes error, which is quite surprising for users coming from Pandas background:
```python
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 !

Contributor guide

Open the contributing guide

Research direction

Start by reproducing the DataFrame and Mars Tensor example in the issue, then trace DataFrame .loc indexing and the tensor-backed index path. Add coverage for df.loc[df.index].execute() and confirm the example completes without the reported error.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
data
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.