覆盖两个熊猫Dataframes或numpy数组并创建一个键值字典

Question

我有两个CSV文件的形状可以说3x3。

文件1看起来像这样：

-1, 2,-1 
-1,-1, 3 
-1, 8, 9 

和文件2是这样的：

-1, 56,-1 
-1,-1, 73 
-1, 24, 100 

我的目标是覆盖两个阵列或DataFrame S和创建以下形式的字典：

dict = {2:56, 3:73, 8:24, 9:100} 

我可以创建一些基于循环的东西，但我想知道是否有办法掩盖或覆盖数组并创建相应的字典？

Answer 1

您可以展平dataframes，放下空值，和字典/ ZIP他们：

import pandas as pd 

# Load the data, converting -1 to NaN 
dfa = pd.read_csv('file1.csv', header=None, na_values=[-1]) 
dfb = pd.read_csv('file2.csv', header=None, na_values=[-1]) 

# Flatten the dataframes 
a = dfa.values.flatten() 
b = dfb.values.flatten() 

# Remove null values and cast back to ints (if that matters) 
# Note that both are filtered according to the key data 
a = a[~np.isnan(a)].astype(int) 
b = b[~np.isnan(a)].astype(int) 

# Zip to pair the lists, then convert to a dict 
d = dict(zip(a, b))