python - Combine elements in list of Tuples? -

-

i'm working on program takes in imdb text file, , outputs top actors (by movie appearances) based on user input n.

however, i'm running issue i'm having slots taken actors in same amount of movies, need avoid. rather, if 2 actors in 5 movies, example, number 5 should appear , actors names should combined , separated semicolon.

i've tried multiple workarounds , nothing has yet worked. suggestions?

if __name__ == "__main__":     imdb_file = raw_input("enter name of imdb file ==> ").strip()     print imdb_file     n= input('enter number of top individuals ==> ')     print n       actors_to_movies = {}      line in open(imdb_file):         words = line.strip().split('|')         actor = words[0].strip()         movie = words[1].strip()         if not actor in actors_to_movies:             actors_to_movies[actor] = set()         actors_to_movies[actor].add(movie)      movie_list= sorted(list(actors_to_movies[actor]))       #arranges dictionary list of tuples#     d = [ (x, actors_to_movies[x]) x in actors_to_movies]     descending = sorted(d, key = lambda x: len(x[1]), reverse=true)      #prints tuples in descending order n number of times (user input)#     in range(n):         print str(len(descending[i][1]))+':', descending[i][0]

there useful method itertools.groupby

it allows split list groups key. using can write function prints top actors:

import itertools def print_top_actors(actor_info_list, top=5):     """     :param: actor_info_list should contain tuples of (actor_name, movie_count)     """     actor_info_list.sort(key=lambda x: x[1], reverse=true)     i, (movie_count, actor_iter) in enumerate(itertools.groupby(actor_info_list)):         if >= top:             break         print movie_count, ';'.join(actor actor, movie_count in actor_iter)

and example of usage:

>>> print_top_actors( ...     [ ...         ("dicaprio", 100500), ...         ("pitt", 100500), ...         ("foo", 10), ...         ("bar", 10), ...         ("baz", 10), ...         ("qux", 3), ...         ("lol", 1) ...     ], top = 3) 100500 dicaprio;pitt 10 foo;bar;baz 3 qux

Comments

Post a Comment

Popular posts from this blog

Django REST Framework perform_create: You cannot call `.save()` after accessing `serializer.data` -

-
i have following viewset: class artistprofileviewset(viewsets.modelviewset): queryset = artistprofile.objects.all() serializer_class = artistprofileserializer def perform_create(self, serializer): if serializer.is_valid(): serializer.save() with following serializers: class simpleartisttrackserializer(serializers.hyperlinkedmodelserializer): class meta: model = artisttrack fields = (...my fields...) class artistprofileserializer(serializers.hyperlinkedmodelserializer): owners = userserializer(many=true, required=false) tracks = simpleartisttrackserializer(many=true, required=false) class meta: model = artistprofile fields = (...my fields...) i getting following error: assertionerror: cannot call `.save()` after accessing `serializer.data`.if need access data before committing database inspect 'serializer.validated_data' instead. i don't see editing serializer.data object. th...
Read more

Why does Go error when trying to marshal this JSON? -

-
i'm trying rather simple example of trying decode json file structs using golang. however, when attempting error that json: cannot unmarshal object go value of type []main.track which don't understand. here's code in question, , json can found here . package main import ( "encoding/json" "fmt" "net/http" "os" ) // tracks group of track types type tracks struct { tracks []track } // track represents singular track type track struct { name string url string artist artist } // artist represents single artist type artist struct { name string mbid string url string } func main() { url := "http://ws.audioscrobbler.com/2.0/?method=geo.gettoptracks&api_key=c1572082105bd40d247836b5c1819623&format=json&country=netherlands" response, err := http.get(url) if err != nil { fmt.printf("%s\n", err) os.exit(1) } de...
Read more