Haskell. INNER JOIN de dos listas
He estado preguntando a google y tratando de encontrar las herramientas necesarias para hacer esto, pero parece que no puedo encontrar una solución a mi problema. Toda la tarea consiste en utilizar un diccionario para traducir una lista y conservar solo los valores únicos. Por ejemplo, las siguientes entradas:
dict = [("a", "aa"), ("b", "bb")]
list1 = ["a","b","c","a","a","g"]
se supone que producen el siguiente resultado:
result = ['aa', 'bb']
Aquí es donde estoy atrapado hasta ahora:
main = do
let dict = [("a", "aa"), ("b", "bb")]
let list1 = ["a","b","c","g"]
let keys = map fst dict
let values = map snd dict
let result = replace keys values list1
print(result)
cuyos rendimientos
["aa","bb","c","g"]
Así que para resolver este estaba pensando que podría haber una manera de utilizar filtero de mapo foldlde alguna manera para hacer una combinación interna de la siguiente manera:
let result = innerJoin values result
Actualmente tengo algo que se parece a esto:
innerJoin :: (Eq a) =>[a] -> [a] -> [a]
innerJoin xs [] = xs
innerJoin [] ys = ys
innerJoin (x:xs) (y:ys) = if (elem x (y:ys))
then x: innerJoin xs ys
else innerJoin xs ys
main = do
let dict = [("a", "aa"), ("b", "bb")]
let list1 = ["a","b","c","g"]
let keys = map fst dict
let values = map snd dict
let list2 = innerJoin keys list1
let result = replace keys values list2
print(result)
Pero vuelve ["aa","bb","c","g"]y no el esperado ["aa","bb"]. Al final, planeo terminarlo nub, pero estoy luchando para descubrir la parte de innerJoin.
EDITAR:
Gracias a la respuesta a continuación, aquí está la solución al problema:
innerJoin xs [] = []
innerJoin [] ys = []
innerJoin (x:xs) (y:ys) = if (elem x (y:ys))
then x: innerJoin xs ys
else innerJoin xs ys
catMaybes ls = [x | Just x <- ls]
genList x [] = []
genList [] y = []
genList x (y:ys) = lookup y x: genList x ys
func dict list = do
let keys = map fst dict
let list1 = innerJoin keys list
catMaybes (genList dict list1)
test1 = func [("a", "aa"),("e", "bb")] ["a","b","c","g","a"]
test2 = func [(1,11),(2,11),(4,44)] [1,2,3,1,2]
Respuestas
La función que desea para esto es lookup. Por ejemplo:
lookup "a" [("a", "aa"), ("b", "bb")] = Just "aa"
lookup "b" [("a", "aa"), ("b", "bb")] = Just "bb"
lookup "x" [("a", "aa"), ("b", "bb")] = Nothing -- "x" is not in the list
La otra función que es útil aquí es Data.Maybe.catMaybes: si tiene una lista como [Just 1, Just 2, Nothing, Just 3, Nothing]y la aplica catMaybes, obtiene [1, 2, 3]. Por lo que puede combinar lookupy catMaybesobtener catMaybes $ map (flip lookup dict) list1como su solución deseada. Un refinamiento adicional es tener en cuenta que Data.Maybedefine mapMaybe f xscomo equivalente a catMaybe (map f xs), por lo que puede simplificarlo a mapMaybe (flip lookup dict) list1.