Sync LeetCode submission Runtime - 414 ms (70.52%), Memory - 60.8 MB (100.00%)

Author: jimit105

2223-the-number-of-rich-customers/README.md

@@ -0,0 +1,46 @@
+<p>Table: <code>Store</code></p>
+
+<pre>
++-------------+------+
+| Column Name | Type |
++-------------+------+
+| bill_id     | int  |
+| customer_id | int  |
+| amount      | int  |
++-------------+------+
+bill_id is the primary key (column with unique values) for this table.
+Each row contains information about the amount of one bill and the customer associated with it.
+</pre>
+
+<p>&nbsp;</p>
+
+<p>Write a solution to report the number of customers who had <strong>at least one</strong> bill with an amount <strong>strictly greater</strong> than <code>500</code>.</p>
+
+<p>The result format is in the following example.</p>
+
+<p>&nbsp;</p>
+<p><strong class="example">Example 1:</strong></p>
+
+<pre>
+<strong>Input:</strong> 
+Store table:
++---------+-------------+--------+
+| bill_id | customer_id | amount |
++---------+-------------+--------+
+| 6       | 1           | 549    |
+| 8       | 1           | 834    |
+| 4       | 2           | 394    |
+| 11      | 3           | 657    |
+| 13      | 3           | 257    |
++---------+-------------+--------+
+<strong>Output:</strong> 
++------------+
+| rich_count |
++------------+
+| 2          |
++------------+
+<strong>Explanation:</strong> 
+Customer 1 has two bills with amounts strictly greater than 500.
+Customer 2 does not have any bills with an amount strictly greater than 500.
+Customer 3 has one bill with an amount strictly greater than 500.
+</pre>

2223-the-number-of-rich-customers/solution.py

@@ -0,0 +1,6 @@
+import pandas as pd
+
+def count_rich_customers(store: pd.DataFrame) -> pd.DataFrame:
+  rich_customers = store[store['amount'] > 500]
+  count = rich_customers['customer_id'].nunique()
+  return pd.DataFrame({'rich_count': [count]})