Как сделать левую, правую и среднюю строку в pandas кадре данных
в кадре pandas, как я могу применить вид excel left ('state', 2), чтобы принимать только первые две буквы. В идеале я хочу научиться использовать левый, правый и средний в области данных. Поэтому для этого конкретного примера нужен эквивалент, а не "трюк".
data = {'state': ['Auckland', 'Otago', 'Wellington', 'Dunedin', 'Hamilton'],
'year': [2000, 2001, 2002, 2001, 2002],
'pop': [1.5, 1.7, 3.6, 2.4, 2.9]}
df = pd.DataFrame(data)
print df
pop state year
0 1.5 Auckland 2000
1 1.7 Otago 2001
2 3.6 Wellington 2002
3 2.4 Dunedin 2001
4 2.9 Hamilton 2002
Я хочу получить следующее:
pop state year StateInitial
0 1.5 Auckland 2000 Au
1 1.7 Otago 2001 Ot
2 3.6 Wellington 2002 We
3 2.4 Dunedin 2001 Du
4 2.9 Hamilton 2002 Ha
Ответы
Ответ 1
Первые две буквы для каждого значения в столбце:
>>> df['StateInitial'] = df['state'].str[:2]
>>> df
pop state year StateInitial
0 1.5 Auckland 2000 Au
1 1.7 Otago 2001 Ot
2 3.6 Wellington 2002 We
3 2.4 Dunedin 2001 Du
4 2.9 Hamilton 2002 Ha
Для последних двух будет df['state'].str[-2:]. Не знаю, что именно вы хотите для среднего, но вы можете применить произвольную функцию к столбцу с помощью метода apply:
>>> df['state'].apply(lambda x: x[len(x)/2-1:len(x)/2+1])
0 kl
1 ta
2 in
3 ne
4 il
