[ARROW-7362] [Python] ListArray.flatten() should take care of slicing offsets

• Type: Bug
• Status: Resolved
• Priority: Minor
• Resolution: Fixed
• Fix Version/s: 0.16.0
• Component/s: C++, Python
• Labels: pull-request-available
• External issue URL: https://github.com/apache/arrow/issues/16993
• Assignee: Zhuo Peng
• Reporter: Zhuo Peng
• Created: 10/Dec/19 06:51
• Updated: 11/Jan/23 07:53
• Resolved: 18/Dec/19 16:57

Description

Currently ListArray.flatten() simply returns the child array. If a ListArray is a slice of another ListArray, they will share the same child array, however the expected behavior (I think) of flatten() should be returning an Array that's a concatenation of all the sub-lists in the ListArray, so the slicing offset should be taken into account.

 

For example:

a = pa.array([[1], [2], [3]])

assert a.flatten().equals(pa.array([1,2,3]))

1. expected:

a.slice(1).flatten().equals(pa.array([2, 3]))

Issue Links

• links to: GitHub Pull Request #6006