在pandas dataframe中創建一個新列作為另一列的函數

Question

我的pandas數據框有一個現有的列“div”，它有一個字符串。 我想創建一個新列（'newcol'），其值等於div中字符串的第一個字符。

我試圖指定這幾種方法，但它不起作用。

results['newcol'] = results['div']給我完整的字符串（如預期的那樣）而不是第一個字符。

results['newcol'] = results['Div'].values[0]和results['newcol'] = results['Div'][0]使每行中的newcol等於'div'的字符串第一排。

results['newcol'] = str(results['Div'])和results['newcol'] = str(results['Div'])[0]將整個['Div']系列轉換為單個字符串將其返回給newcol。

指定我想要的正確方法是什么？

Answer 1

這應該工作：

import pandas as pd
data = pd.DataFrame({"A": ["hello", "world"], "B": [1, 2]})
data["C"] = data.A.str[0]
data

這是輸出：

  |   A   | B | C 
------------------
0 | hello | 1 | h
------------------
1 | world | 2 | w