How would I use lens in Haskell to duplicate Python's enumerate?

后端 未结 3 1691
孤独总比滥情好
孤独总比滥情好 2021-01-13 19:37

Python\'s enumerate on lists can be written as zip [0..]. I looked at Control.Lens.Traversal and Control.Lens.Indexed, but I couldn\'t figure out how to use len

3条回答
  •  迷失自我
    2021-01-13 20:08

    If you're using a container that is an instance of FunctorWithIndex then you can simply use imap (,):

    > imap (,) "abc"
    [(0,'a'),(1,'b'),(2,'c')]
    

    But if the index isn't the position this won't work:

    > let m = Map.fromList [('a', "foo"), ('b', "bar"), ('c', "foobar")])
    > imap (,) m
    fromList [('a',('a',"foo")),('b',('b',"bar")),('c',('c',"foobar"))]
    

    Instead you can use traversed, which is an indexed traversal where the index is the order the elements appear. This can be used for anything that's Traversable. Instead of imap use iover traversed (which is the same as imapOf but that's been deprecated):

    > iover traversed (,) "abc"
    [(0,'a'),(1,'b'),(2,'c')]
    
    > iover traversed (,) m
    fromList [('a',(0,"foo")),('b',(1,"bar")),('c',(2,"foobar"))]
    

提交回复
热议问题