Sorted Sets in Redis

Welcome back! Building on our previous experience with Redis sets, today we are diving into sorted sets. Redis sorted sets combine the power of sets and lists, allowing us to handle collections in which every member is unique but has an associated score. These scores ensure the elements are kept in a specific, sorted order.

In this lesson, you will understand how to use sorted sets in Redis. Specifically, we will focus on:

1. Adding members and scores to a sorted set.
2. Retrieving top members based on their scores.

Sorted sets in Redis are remarkable due to their efficiency and flexibility. You might find them particularly useful for scenarios like maintaining leaderboards, scheduling tasks, or storing time-series data.

Let’s start by connecting to your Redis server and adding some members to a sorted set:

import redis

# Connect to Redis
client = redis.Redis(host='localhost', port=6379, db=0)

# Adding scores and members to a sorted set
client.zadd('leaderboard', {'Alice': 100, 'Bob': 400, 'Charlie': 300, 'Alice': 350})

# Retrieving top players
top_players = client.zrevrange('leaderboard', 0, 1, withscores=True)
top_players = [(member.decode('utf-8'), score) for member, score in top_players]
print(f"Top 2 players: {top_players}")

# Retrieve players with lowest scores
low_players = client.zrange('leaderboard', 0, 1, withscores=True)
low_players = [(member.decode('utf-8'), score) for member, score in low_players]
print(f"Lowest 2 players: {low_players}")

This code works by using the zadd command to add members with their scores and the zrevrange command to get members in descending order of their scores.

Notice how we used the withscores parameter to include scores in the result. This way, we can easily retrieve the top players along with their scores.

For this example, the output will be Top 2 players: [('Bob', 400.0), ('Alice', 350.0)]. Notice, that the score for Alice is 350.0, not 100.0, as the last score is the one that is kept.

Similarly, you can use the zrange command to retrieve members in ascending order of their scores. This is useful when you want to get the lowest scores.

Now, let's also learn how to remove members from a sorted set:

# Removing members from a sorted set
client.zrem('leaderboard', 'Alice')

In this code snippet, we used the zrem command to remove the member 'Alice' from the 'leaderboard' sorted set.