Как показано в твоём примере работать не будет.
Я модифицировал твой пример - так работает:
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<meta http-equiv="X-UA-Compatible" content="ie=edge">
<title>Document</title>
</head>
<body>
<div class='d-flex'>
<div class='container d-flex flex-wrap'>
<button>1</button>
<button>2</button>
<button>3</button>
<button class="other">5</button>
<button>6</button>
</div>
<div class='d-flex flex-wrap'>
<button>4</button>
</div>
<div class='container'>
</body>
</html>
.d-flex {
display: flex;
}
.flex-wrap {
flex-wrap: wrap;
}
.container {
width: auto;
max-width: 240px;
}
button {
width: 80px;
}
.other {
width: 160px;
}