英文:
To find a column name in a specific column and enter its number
问题
Python.
我将使用pandas。
从第1列中定位列名。
在输入列0中找到文本中的列名，并将其输入为数字。
import pandas as pddata = [['EA', 0, 0, 0], ['col2,col2', 0, 0, 0], ['col3', 0, 0, 0]]df_input = pd.DataFrame(data, columns=[0, 'EA', 'col2', 'col3'])df_inputdata = [['EA', 1, 0, 0], ['col2,col2', 0, 2, 0], ['col3', 0, 0, 1]]df_output = pd.DataFrame(data, columns=[0, 'EA', 'col2', 'col3'])df_output
英文:
Python.
I'm going to use a pandas
Locate the column name from column 1.
Locate the column name in the text entered in column 0 and enter the number as a number.
import pandas as pddata = [['EA',0,0,0], ['col2,col2', 0, 0,0], ['col3', 0,0,0]]df_input = pd.DataFrame(data, columns=[0, 'EA', 'col2', 'col3'])df_inputdata = [['EA',1,0,0], ['col2,col2', 0, 2,0], ['col3', 0,0,1]]df_output = pd.DataFrame(data, columns=[0, 'EA', 'col2', 'col3'])df_output
答案1
得分: 2
你可以使用 str.extractall，然后使用 crosstab 进行操作：
s = (df_input[0].str.extractall('([^,]+)')[0].droplevel(-1))# 在原地更新df_input.update(pd.crosstab(s.index, s))
或者创建一个新的 DataFrame：
df_output = (pd.crosstab(s.index, s).combine_first(df_input).reindex_like(df_input))
输出如下：
0 EA col2 col30 EA 1 0 01 col2,col2 0 2 02 col3 0 0 1
如果 col2,col2 变成了 col2,col3，输出如下：
0 EA col2 col30 EA 1 0 01 col2,col3 0 1 12 col3 0 0 1
英文:
You can use str.extractall then a crosstab:
s = (df_input[0].str.extractall('([^,]+)')[0].droplevel(-1))# update in placedf_input.update(pd.crosstab(s.index, s))
Or for a new DataFrame:
df_output = (pd.crosstab(s.index, s).combine_first(df_input).reindex_like(df_input))
Output:
0 EA col2 col30 EA 1 0 01 col2,col2 0 2 02 col3 0 0 1
Output if col2,col2 was col2,col3:
0 EA col2 col30 EA 1 0 01 col2,col3 0 1 12 col3 0 0 1
通过集体智慧和协作来改善编程学习和解决问题的方式。致力于成为全球开发者共同参与的知识库，让每个人都能够通过互相帮助和分享经验来进步。
评论