IS
Помогите мне плиз, можете ответить анонимно на 6 вопросов про сообщество?
Собираю сегодня до 22:00 мск.
Если интересно, потом поделюсь статистикой =)
https://forms.gle/WEUWig4C6bKDqk9L7
Size: a a a
IS
J
I
E
df = pd.DataFrame({0: [1,5,2], 1: [4,2,3]}, index=list('abc'))То есть нужно вытащить в новую таблицу один столбец по argmax из исходной таблицы, для каждой строки.
0 1
a 1 4
b 5 2
c 2 3
# получаем серию с максимальным столбцом для каждой строки
m = df.idxmax(axis=1)
a 1
b 0
c 1
# делаем некую индексацию и получаем такой датафрейм с одним столбцом:
0
a 4
b 5
c 3
a = np.array([(1,4),(5,2),(2,3)])
array([[1, 4],
[5, 2],
[2, 3]])
m = a.argmax(axis=1)
array([1, 0, 1], dtype=int64)
b = a[range(a.shape[0]), m]
array([4, 5, 3])
N
df = pd.DataFrame({0: [1,5,2], 1: [4,2,3]}, index=list('abc'))То есть нужно вытащить в новую таблицу один столбец по argmax из исходной таблицы, для каждой строки.
0 1
a 1 4
b 5 2
c 2 3
# получаем серию с максимальным столбцом для каждой строки
m = df.idxmax(axis=1)
a 1
b 0
c 1
# делаем некую индексацию и получаем такой датафрейм с одним столбцом:
0
a 4
b 5
c 3
a = np.array([(1,4),(5,2),(2,3)])
array([[1, 4],
[5, 2],
[2, 3]])
m = a.argmax(axis=1)
array([1, 0, 1], dtype=int64)
b = a[range(a.shape[0]), m]
array([4, 5, 3])
E
N
N
E
1 0 1
a 4 1 4
b 2 5 2
c 3 2 3
df.iloc[range(m.size), m]
E
take_along_axis
, с которой можно сделать то же самое примерно:np.take_along_axis(a, m[:, np.newaxis], axis=1)
array([[4],
[5],
[3]])
E
df.lookup(m.index, m)
array([4, 5, 3], dtype=int64)
N
AB
I
AB
I
AB
I
I
I