Geek Answers Handbook

Pandas vector data fetch

I have a simple df forming pivot_table:

    d = {'one' : ['A', 'B', 'B', 'C', 'C', 'C'], 'two' : [6., 5., 4., 3., 2., 1.],     'three' : [6., 5., 4., 3., 2., 1.], 'four' : [6., 5., 4., 3., 2., 1.]}
    df = pd.DataFrame(d)
    pivot = pd.pivot_table(df,index=['one','two'])

I would like to randomly sample 1 row from each other element from the column "one" of the resulting rotation object. (In this example, “A” will always be selected, while there are more options for “B” and “C”.) I just started using pandas version 0.18.0 and I know about .sample . I messed up the .groupby method using the fetch function like this:

    grouped = pivot.groupby('one').apply(lambda x: x.sample(n=1, replace=False))

I raise KeyError when I tried variations of this theme, so I thought it was time for a new perspective on this seemingly simple question ...

Thanks for any help!

+4
source share
1

KeyError , "" pivot, :

In [11]: pivot
Out[11]:
         four  three
one two
A   6.0   6.0    6.0
B   4.0   4.0    4.0
    5.0   5.0    5.0
C   1.0   1.0    1.0
    2.0   2.0    2.0
    3.0   3.0    3.0

:

In [12]: pivot.groupby(level='one').apply(lambda x: x.sample(n=1, replace=False))
Out[12]:
             four  three
one one two
A   A   6.0   6.0    6.0
B   B   4.0   4.0    4.0
C   C   1.0   1.0    1.0

, ! as_index=False:

In [13]: pivot.groupby(level='one', as_index=False).apply(lambda x: x.sample(n=1))
Out[13]:
           four  three
  one two
0 A   6.0   6.0    6.0
1 B   4.0   4.0    4.0
2 C   2.0   2.0    2.0

. .

, ( :

In [21]: df.iloc[[np.random.choice(x) for x in g.indices.values()]]
Out[21]:
   four one  three  two
1   5.0   B    5.0  5.0
3   3.0   C    3.0  3.0
0   6.0   A    6.0  6.0
+3

More articles:


All Articles