制作成对的文件和列以供循环使用？

Question

我想制作一对文件和循环中需要的列。

我想一本字典可以，但不确定。

例子：

files: file1,file2,file3,etc

dict1 = ({file1:its needed columns,file2:its needed columns})

后面会用到这个函数：

for i in dict1: # below it reads the files from arcpy - not important 

    df = pd.DataFrame.from_records(data=arcpy.da.SearchCursor("key_of_dict",

                                                     ['the_one_column','the_other_column'])

    #then make a new column that will apply the value_counts to a certain column

    df['count_of_a_col']=[df['one_col'].value_counts().loc[x] for x in df['one_col']]

我怎样才能使这项工作？

笔记

每个文件中的列并不总是相同的。对于一个文件，我们需要两个特定的列，而对于另一个则完全不同。这就是我考虑使用字典的原因。

Answer 1

你在找这个吗？dict1 = {'file_name_1': ['on_column', 'another_column'], 'file_name_2': ['again_column']}for k, v in dict1.items():    df = pd.DataFrame.from_records(data=arcpy.da.SearchCursor(k, v))    column_for_count = v[1] if len(v) >= 2 else v[0]    df['count_of_' + column_for_count]=[df[column_for_count].value_counts().loc[x] for x in df[column_for_count]]    # do what you want with v