What are levels in a pandas DataFrame?

前端 未结 2 1217
天命终不由人
天命终不由人 2020-12-14 02:45

I\'ve been reading through the documentation and many explanations and examples use levels as something taken for granted. Imho the docs lack a bit on a fundame

相关标签:
2条回答
  • 2020-12-14 03:23

    Usually a DataFrame has a 1D index and columns:

        x y
    0   4 1
    1   3 9
    

    Here the index is [0, 1] and the columns are ['x', 'y']. But you can have multiple levels in either the index or the columns:

        x y
        a b c
    0 7 4 1 3
      8 3 9 5
    

    Here the columns' first level is ['x', 'y', 'y'] and the second level is ['a', 'b', 'c']. The index's first level is [0, 0] and the second level is [7, 8].

    0 讨论(0)
  • 2020-12-14 03:26

    I stumbled across this question while analyzing the answer to my own question, but I didn't find the John's answer satisfying enough. After a few experiments though I think I understood the levels and decided to share:

    Short answer:

    Levels are parts of the index or column.

    Long answer:

    I think this multi-column DataFrame.groupby example illustrates the index levels quite nicely.

    Let's say we have the time logged on issues report data:

    report = pd.DataFrame([
            [1, 10, 'John'],
            [1, 20, 'John'],
            [1, 30, 'Tom'],
            [1, 10, 'Bob'],
            [2, 25, 'John'],
            [2, 15, 'Bob']], columns = ['IssueKey','TimeSpent','User'])
    
       IssueKey  TimeSpent  User
    0         1         10  John
    1         1         20  John
    2         1         30   Tom
    3         1         10   Bob
    4         2         25  John
    5         2         15   Bob
    

    The index here has only 1 level (there is only one index value identifying every row). The index is artificial (running number) and consists of values form 0 to 5.

    Say we want to merge (sum) all logs created by the same user to the same issue (to get the total time spent on the issue by the user)

    time_logged_by_user = report.groupby(['IssueKey', 'User']).TimeSpent.sum()
    
    IssueKey  User
    1         Bob     10
              John    30
              Tom     30
    2         Bob     15
              John    25
    

    Now our data index has 2 levels, as multiple users logged time to the same issue. The levels are IssueKey and User. The levels are parts of the index (only together they can identify a row in a DataFrame / Series).

    Levels being parts of the index (as a tuple) can be nicely observed in the Spyder Variable explorer:

    Having levels gives us opportunity to aggregate values within groups in respect to an index part (level) of our choice. E.g. if we want to assign the max time spent on an issue by any user, we can:

    max_time_logged_to_an_issue = time_logged_by_user.groupby(level='IssueKey').transform('max')
    
    IssueKey  User
    1         Bob     30
              John    30
              Tom     30
    2         Bob     25
              John    25
    

    Now the first 3 rows have the value 30, as they correspond to the issue 1 (User level was ignored in the code above). The same story for the issue 2.

    This can be useful e.g. if we want to find out which users spent most time on every issue:

    issue_owners = time_logged_by_user[time_logged_by_user == max_time_logged_to_an_issue]
    
    IssueKey  User
    1         John    30
              Tom     30
    2         John    25
    
    0 讨论(0)
提交回复
热议问题