Workspace
James Clunies-Ross/

Experiment with the AI assistant

0
Beta
Spinner

Experiment with the AI assistant

Generative AI is coming to DataCamp Workspace! Go through this notebook to experience the AI-enabled functionality. In this example project, we'll be querying a database with unicorn company data to build a chart showing the countries with the highest amount in total funding.

Get the data

The first step is getting data. Rather than using SQL, we'll ask the AI assistant to generate the code for us!

  • Click on the SQL cell below and select "Unicorn Companies" in the dropdown in the cell header.
  • In the cell menu on the right-hand side, click on "Generate".
  • In the input field that appears, type "Get a list of companies (name, country, and funding)".
  • Hit Enter and see the code streaming in!
  • Accept the suggestion and run the SQL cell.
Unknown integration
DataFrameavailable as
df
variable
SELECT set_num, name, num_parts
FROM sets
ORDER BY num_parts DESC
LIMIT 10;
This query is taking long to finish...Consider adding a LIMIT clause or switching to Query mode to preview the result.

Visualize the data

To visualize the DataFrame we got back in the previous step, let's ask the AI once more!

  • Click in the Python cell below and click on "Generate" in the cell menu on the right-hand side.
  • In the input field that appears, type "Plotly bar chart with top 10 countries by total funding".
  • Hit Enter and see the code streaming in!
  • Accept the suggestion and run the code cell.
import plotly.express as px

'''
Sort the dataframe by total funding in descending order
'''

df_sorted = df.groupby('country')['funding'].sum().reset_index().sort_values('funding', ascending=False)

# Select the top 10 countries
top_10_countries = df_sorted.head(10)

# Create the bar chart using Plotly
fig = px.bar(top_10_countries, x='country', y='funding', title='Top 10 Countries by Total Funding')

# Show the chart
fig.show()

May the AI force be with you! 🤖

  • AI Chat
  • Code